OT Primitive Universal Flow / results.json

Mechanism confirmed, baseline not beaten

Raw ⬇ ZIP
 1{
 2  "device": "cuda",
 3  "sanity_check": {
 4    "gaussian_mean_error": 0.012130946852266788,
 5    "gaussian_covariance_error": 0.027502108365297318,
 6    "translation_cost_scaling": [
 7      {
 8        "L": 1,
 9        "predicted": 3.125,
10        "observed": 3.125,
11        "relative_error": 0.0
12      },
13      {
14        "L": 2,
15        "predicted": 1.5625,
16        "observed": 1.5625,
17        "relative_error": 0.0
18      },
19      {
20        "L": 4,
21        "predicted": 0.78125,
22        "observed": 0.78125,
23        "relative_error": 0.0
24      },
25      {
26        "L": 8,
27        "predicted": 0.390625,
28        "observed": 0.390625,
29        "relative_error": 0.0
30      },
31      {
32        "L": 16,
33        "predicted": 0.1953125,
34        "observed": 0.1953125,
35        "relative_error": 0.0
36      }
37    ]
38  },
39  "mini_experiment": {
40    "baseline_unconstrained_residual": {
41      "mmd": 0.011374211870133877,
42      "sliced_wasserstein": 0.20292143523693085,
43      "displacement_cost": 0.30849307775497437,
44      "jacobian_positive_fraction": 1.0,
45      "jacobian_min": 0.6527191996574402,
46      "jacobian_max": 26.707469940185547
47    },
48    "ot_gradient_potential": {
49      "mmd": 0.011317086406052113,
50      "sliced_wasserstein": 0.2003474235534668,
51      "displacement_cost": 0.30494531989097595,
52      "jacobian_positive_fraction": 1.0,
53      "jacobian_min": 1.580175518989563,
54      "jacobian_max": 26.42418670654297
55    }
56  },
57  "settings": {
58    "seed": 2359,
59    "samples": 192,
60    "layers": 2,
61    "steps": 260,
62    "mmd_weight": 12
63  }
64}