Create a polynomial chaos metamodel

In this example we are going to create a global approximation of a model response using functional chaos.

Let h be the function defined by:

g(\mathbf{x}) = \left[\cos(x_1 + x_2), (x_2 + 1) e^{x_1}\right]

for any \mathbf{x}\in\mathbb{R}^2.

We assume that

X_1 \sim \mathcal{N}(0,1) \textrm{ and } X_2 \sim \mathcal{N}(0,1)

and that X_1 and X_2 are independent.

An interesting point in this example is that the output is multivariate. This is why we are going to use the getMarginal method in the script in order to select the output marginal that we want to manage.

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

We first create the function model.

ot.RandomGenerator.SetSeed(0)
dimension = 2
input_names = ['x1', 'x2']
formulas = ['cos(x1 + x2)', '(x2 + 1) * exp(x1)']
model = ot.SymbolicFunction(input_names, formulas)

Then we create a sample inputSample and compute the corresponding output sample outputSample.

distribution = ot.Normal(dimension)
samplesize = 80
inputSample = distribution.getSample(samplesize)
outputSample = model(inputSample)

Create a functional chaos model. First, we need to fit a distribution on the input sample. We can do this automatically with the Lilliefors test.

ot.ResourceMap.SetAsUnsignedInteger(
    "FittingTest-LillieforsMaximumSamplingSize", 100)
algo = ot.FunctionalChaosAlgorithm(inputSample, outputSample)
algo.run()
result = algo.getResult()
metamodel = result.getMetaModel()

Plot the second output of our model depending on x_2 with x_1=0.5. In order to do this, we create a ParametricFunction and set the value of x_1. Then we use the getMarginal method to extract the second output (which index is equal to 1).

x1index = 0
x1value = 0.5
x2min = -3.
x2max = 3.
outputIndex = 1
metamodelParametric = ot.ParametricFunction(metamodel, [x1index], [x1value])
graph = metamodelParametric.getMarginal(outputIndex).draw(x2min, x2max)
graph.setLegends(["Metamodel"])
modelParametric = ot.ParametricFunction(model, [x1index], [x1value])
curve = modelParametric.getMarginal(
    outputIndex).draw(x2min, x2max).getDrawable(0)
curve.setColor('red')
curve.setLegend("Model")
graph.add(curve)
graph.setLegendPosition("bottomright")
graph.setXTitle("X2")
graph.setTitle("Metamodel Validation, output #%d" % (outputIndex))
view = viewer.View(graph)
Metamodel Validation, output #1

We see that the metamodel fits approximately to the model, except perhaps for extreme values of x_2. However, there is a better way of globally validating the metamodel, using the MetaModelValidation on a validation design of experiment.

n_valid = 100
inputTest = distribution.getSample(n_valid)
outputTest = model(inputTest)

Plot the corresponding validation graphics.

val = ot.MetaModelValidation(inputTest, outputTest, metamodel)
Q2 = val.computePredictivityFactor()
graph = val.drawValidation()
graph.setTitle("Metamodel validation Q2="+str(Q2))
view = viewer.View(graph)
Metamodel validation Q2=[0.775103,0.889995]

The coefficient of predictivity is not extremely satisfactory for the first output, but is would be sufficient for a central dispersion study. The second output has a much more satisfactory Q2: only one single extreme point is far from the diagonal of the graphics.

Compute and print Sobol’ indices

chaosSI = ot.FunctionalChaosSobolIndices(result)
print(chaosSI.summary())

Out:

 input dimension: 2
 output dimension: 2
 basis size: 18
 mean: [0.380829,1.62643]
 std-dev: [0.686114,3.48991]
------------------------------------------------------------
Marginal: 0
Index   | Multi-indice                  | Part of variance
------------------------------------------------------------
     16 | [2,2]                         | 0.306305
     15 | [1,3]                         | 0.18002
     14 | [3,1]                         | 0.177155
      4 | [0,2]                         | 0.130582
      3 | [2,0]                         | 0.0962271
      5 | [1,1]                         | 0.0610813
      7 | [0,4]                         | 0.0307299
      6 | [4,0]                         | 0.0178984
------------------------------------------------------------


------------------------------------------------------------
Component | Sobol index            | Sobol total index
------------------------------------------------------------
        0 | 0.114126               | 0.838688
        1 | 0.161312               | 0.885874
------------------------------------------------------------

Marginal: 1
Index   | Multi-indice                  | Part of variance
------------------------------------------------------------
      5 | [1,1]                         | 0.288059
      2 | [0,1]                         | 0.271055
      1 | [1,0]                         | 0.156647
     10 | [2,1]                         | 0.130254
      3 | [2,0]                         | 0.0984672
     14 | [3,1]                         | 0.0322146
      8 | [5,0]                         | 0.0172213
------------------------------------------------------------


------------------------------------------------------------
Component | Sobol index            | Sobol total index
------------------------------------------------------------
        0 | 0.278367               | 0.728896
        1 | 0.271104               | 0.721633
------------------------------------------------------------

Let us analyse the results of this global sensitivity analysis.

  • We see that the first output involves significant multi-indices with total degree 4. The contribution of the interactions are very significant in this model.

  • The second output involves multi-indices with total degrees from 1 to 7, with a significant contribution of multi-indices with total degress 5 and 7. The first variable is especially significant, with a significant contribution of the interactions.

Draw Sobol’ indices.

sensitivityAnalysis = ot.FunctionalChaosSobolIndices(result)
first_order = [sensitivityAnalysis.getSobolIndex(i) for i in range(dimension)]
total_order = [sensitivityAnalysis.getSobolTotalIndex(
    i) for i in range(dimension)]
input_names = model.getInputDescription()
graph = ot.SobolIndicesAlgorithm.DrawSobolIndices(
    input_names, first_order, total_order)
graph.setLegendPosition("center")
view = viewer.View(graph)
Sobol' indices

Testing the sensitivity to the degree

With the specific constructor of FunctionalChaosAlgorithm that we use, the FunctionalChaosAlgorithm-MaximumTotalDegree in the ResourceMap configure the maximum degree explored by the algorithm. This degree is a trade-off.

  • If the maximum degree is too low, the polynomial may miss some coefficients so that the quality is lower than possible.

  • If the maximum degree is too large, the number of coefficients to explore is too large, so that the coefficients might be poorly estimated.

This is why the following for loop explores various degrees to see the sensitivity of the metamodel predictivity depending on the degree.

The default value of this parameter is 10.

ot.ResourceMap.GetAsUnsignedInteger(
    "FunctionalChaosAlgorithm-MaximumTotalDegree")

Out:

10

This is why we explore the values from 5 to 15.

degrees = range(5, 12)
q2 = ot.Sample(len(degrees), 2)
for maximumDegree in degrees:
    ot.ResourceMap.SetAsUnsignedInteger(
        "FunctionalChaosAlgorithm-MaximumTotalDegree", maximumDegree)
    print("Maximum total degree =", maximumDegree)
    algo = ot.FunctionalChaosAlgorithm(inputSample, outputSample)
    algo.run()
    result = algo.getResult()
    metamodel = result.getMetaModel()
    for outputIndex in range(2):
        val = ot.MetaModelValidation(
            inputTest, outputTest[:, outputIndex], metamodel.getMarginal(outputIndex))
        q2[maximumDegree - degrees[0],
            outputIndex] = val.computePredictivityFactor()[0]

Out:

Maximum total degree = 5
Maximum total degree = 6
Maximum total degree = 7
Maximum total degree = 8
Maximum total degree = 9
Maximum total degree = 10
Maximum total degree = 11
graph = ot.Graph("Predictivity", "Total degree", "Q2", True)
cloud = ot.Cloud([[d] for d in degrees], q2[:, 0])
cloud.setLegend("Output #0")
cloud.setPointStyle("bullet")
graph.add(cloud)
cloud = ot.Cloud([[d] for d in degrees], q2[:, 1])
cloud.setLegend("Output #1")
cloud.setColor("red")
cloud.setPointStyle("bullet")
graph.add(cloud)
graph.setLegendPosition("topright")
view = viewer.View(graph)
plt.show()
Predictivity

We see that a total degree lower than 9 is not sufficient to describe the first output with good predictivity. However, the coefficient of predictivity drops when the total degree gets greater than 12. The predictivity of the second output seems to be much less satisfactory: a little more work would be required to improve the metamodel.

In this situation, the following methods may be used.

  • Since the distribution of the input is known, we may want to give this information to the FunctionalChaosAlgorithm. This prevents the algorithm from trying to fit the distribution which best fit to the data.

  • We may want to customize the adaptiveStrategy by selecting an enumerate function which best fit to this particular situation. In this specific example, the interactions plays a great role so that the linear enumerate function may provide better results than the hyperbolic rule.

  • We may want to customize the projectionStrategy by selecting an method to compute the coefficient which improves the estimation. Given that the function is symbolic and fast, it might be interesting to try an integration rule instead of the least squares method.

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

Gallery generated by Sphinx-Gallery