2.4.8 SDE with Stratonovich interpretation

This example solves a stochastic diffusion equation with Stratonovich interpretation

  0%|          | 0/10.0 [00:00<?, ?it/s]
Initializing:   0%|          | 0/10.0 [00:00<?, ?it/s]
  0%|          | 0/10.0 [00:04<?, ?it/s]
  0%|          | 0.005/10.0 [00:04<2:29:02, 894.66s/it]
  2%|▏         | 0.159/10.0 [00:04<04:38, 28.27s/it]
 12%|█▏        | 1.198/10.0 [00:04<00:34,  3.88s/it]
 39%|███▉      | 3.9/10.0 [00:05<00:07,  1.29s/it]
 83%|████████▎ | 8.261/10.0 [00:05<00:01,  1.46it/s]
 83%|████████▎ | 8.261/10.0 [00:05<00:01,  1.40it/s]
100%|██████████| 10.0/10.0 [00:05<00:00,  1.70it/s]
100%|██████████| 10.0/10.0 [00:05<00:00,  1.70it/s]

from pde import PDE, ScalarField, UnitGrid


class AllenCahnNoisePDE(PDE):
    """Allen–Cahn PDE with custom noise implementation."""

    use_noise_variance = True

    def make_noise_variance(self, state, *, backend, ret_diff=False):
        """Make function that calculates noise variance."""
        noise = float(self.noise)

        if ret_diff:

            def noise_variance(state_data, t):
                return noise * state_data**2, 2 * noise * state_data

        else:

            def noise_variance(state_data, t):
                return noise * state_data**2

        return noise_variance


eq = AllenCahnNoisePDE(
    rhs={"c": "laplace(c) + c - c**3"}, noise=1.0, noise_interpretation="stratonovich"
)
state = ScalarField.random_uniform(UnitGrid([64, 64]), -1, 1)
result = eq.solve(state, t_range=10, dt=1e-3, solver="milstein")
result.plot()