Estimate Sobol’ indices for the Ishigami function by a sampling method: a quick start guide to sensitivity analysis

In this example, we estimate the Sobol’ indices for the Ishigami function by sampling methods.

Introduction

In this example we are going to quantify the correlation between the input variables and the output variable of a model thanks to Sobol indices.

Sobol indices are designed to evaluate the importance of a single variable or a specific set of variables. Here the Sobol indices are estimated by sampling from the distributions of the input variables and propagating uncertainty through a function.

In theory, Sobol indices range from 0 to 1; the closer an index value is to 1, the better the associated input variable explains the function output.

Let us denote by d the input dimension of the model.

Sobol’ indices can be computed at different orders.

  • First order indices evaluate the importance of one input variable at a time.

  • Total indices give the relative importance of one input variable and all its interactions with other variables. Alternatively, they can be viewed as measuring how much wriggle room remains to the output when all but one input variables are fixed.

  • In general, we are only interested in first order and total Sobol’ indices. There are situations, however, where we want to estimate interactions. Second order indices evaluate the importance of every pair of input variables. The number of second order indices is:

\binom{d}{2} = \frac{d \times \left( d-1\right)}{2}.

In practice, when the number of input variables is not small (say, when d>5), then the number of second order indices is too large to be easily analyzed. In these situations, we limit the analysis to the first order and total Sobol’ indices.

Define the model

import openturns as ot
import numpy as np
import pylab as pl
import openturns.viewer
import openturns.viewer as viewer
from matplotlib import pylab as plt
ot.Log.Show(ot.Log.NONE)

We load the Ishigami model from the usecases model :

from openturns.usecases import ishigami_function
im = ishigami_function.IshigamiModel()

The IshigamiModel data class contains the input distribution X=(X_1, X_2, X_3) in im.distributionX and the Ishigami function in im.model. We also have access to the input variable names with

input_names = im.distributionX.getDescription()

Draw the function

n = 10000
sampleX = im.distributionX.getSample(n)
sampleY = im.model(sampleX)
def plotXvsY(sampleX, sampleY, figsize=(15,3)):
    dimX = sampleX.getDimension()
    inputdescr = sampleX.getDescription()
    fig = pl.figure(figsize=figsize)
    for i in range(dimX):
        ax = fig.add_subplot(1, dimX, i+1)
        graph = ot.Graph('', inputdescr[i], 'Y', True, '')
        cloud = ot.Cloud(sampleX[:,i],sampleY)
        graph.add(cloud)
        _ = ot.viewer.View(graph, figure=fig, axes=[ax])
    return fig

plotXvsY(sampleX, sampleY, figsize=(10,4));
plot sensitivity sobol

Out:

<Figure size 1000x400 with 3 Axes>
graph = ot.HistogramFactory().build(sampleY).drawPDF()
view = viewer.View(graph)
y0 PDF

We see that the distribution of the output has two modes.

Estimate the Sobol’ indices

We first create a design of experiments with the SobolIndicesExperiment. Since we are not interested in second order indices for the moment, we use the default value of the third argument (we will come back to this topic later).

size = 1000
sie = ot.SobolIndicesExperiment(im.distributionX, size)
inputDesign = sie.generate()
input_names = im.distributionX.getDescription()
inputDesign.setDescription(input_names)
inputDesign.getSize()

Out:

5000

We see that 5000 function evaluations are required to estimate the first order and total Sobol’ indices.

Then we evaluate the outputs corresponding to this design of experiments.

outputDesign = im.model(inputDesign)

Then we estimate the Sobol’ indices with the SaltelliSensitivityAlgorithm.

sensitivityAnalysis = ot.SaltelliSensitivityAlgorithm(inputDesign, outputDesign, size)

The getFirstOrderIndices and getTotalOrderIndices method respectively return estimates of all first order and total Sobol’ indices.

sensitivityAnalysis.getFirstOrderIndices()

[0.262063,0.403644,0.0435311]



sensitivityAnalysis.getTotalOrderIndices()

[0.529997,0.441213,0.27389]



The draw method produces the following graph. The vertical bars represent the 95% confidence intervals of the estimates.

graph = sensitivityAnalysis.draw()
view = viewer.View(graph)
Sobol' indices - SaltelliSensitivityAlgorithm
  • We see that the variable X_1, with a total Sobol’ index close to 0.6, is the most significant variable, taking into account both its direct effect and its interactions with other variables. Its first order index is close to 0.3, which implies that its interactions alone produce almost 30% (0.6 - 0.3) of the total variance.

  • The variable X_2 has the highest first order index: approximately 0.4. However, it has little interaction with other variables since its total order indice is close to its first order index.

  • The variable X_3 has a first order index close to zero. However, it has an impact to the total variance thanks to its interactions with X_1.

  • We see that the variability of the estimates is quite high even with a relatively large sample size. Moreover, since the exact first order Sobol’ index for X_3 is zero, its estimate has a 50% chance of being nonpositive.

Estimate the second order indices

size = 1000
computeSecondOrder = True
sie = ot.SobolIndicesExperiment(im.distributionX, size, computeSecondOrder)
inputDesign = sie.generate()
print(inputDesign.getSize())
inputDesign.setDescription(input_names)
outputDesign = im.model(inputDesign)

Out:

8000

We see that 8000 function evaluations are now required; that is 3000 more evaluations than in the previous situation.

sensitivityAnalysis = ot.SaltelliSensitivityAlgorithm(inputDesign, outputDesign, size)
second_order = sensitivityAnalysis.getSecondOrderIndices()
for i in range(im.dim):
    for j in range(i):
        print('2nd order indice (%d,%d)=%g' % (i,j,second_order[i,j]))

Out:

2nd order indice (1,0)=0.0120302
2nd order indice (2,0)=0.217984
2nd order indice (2,1)=0.0590756

This shows that the only significant interaction is the one between X_1 and X_3 (beware of Python’s index shift: 0 denotes the first input variable).

Using a different estimator

We have used the SaltelliSensitivityAlgorithm class to estimate the indices. Others are available in the library:

  • SaltelliSensitivityAlgorithm

  • MartinezSensitivityAlgorithm

  • JansenSensitivityAlgorithm

  • MauntzKucherenkoSensitivityAlgorithm

In order to compare the results with another method, we use the MartinezSensitivityAlgorithm class.

sensitivityAnalysis = ot.MartinezSensitivityAlgorithm(inputDesign, outputDesign, size)
graph = sensitivityAnalysis.draw()
view = viewer.View(graph)

plt.show()
Sobol' indices - MartinezSensitivityAlgorithm

We see that the results do not change significantly in this particular situation.

Total running time of the script: ( 0 minutes 0.511 seconds)

Gallery generated by Sphinx-Gallery