fix: the ConvexAdam presets run the settings ConvexAdam runs
Browse filesThe two presets carried their own tuning -- a fifth of the step size over nearly
twice the iterations, a coarser Adam grid, a lighter regulariser -- and the
refinement ran on an unsmoothed control grid, which no preset could change until
konfai exposed the setting. They now take the reference's own values: grid 6,
half-width 4, 80 iterations at step 1.0, lambda 1.25, the Adam grid at half the
image, and the three smoothing passes the method applies at every iteration.
Measured against the reference implementation on four ThoraxCBCT pairs (mean Dice
over 57 structures, 0.369 unregistered): 0.471 for convexAdam's own defaults,
0.450 for these settings here, 0.424 once the preset's linear pre-alignment runs
first. The pre-alignment is kept, being what makes the preset usable on a pair
that starts far apart, and it is the one setting left to revisit.
Needs konfai 1.8.6 for control_grid_smoothing.
|
@@ -15,12 +15,13 @@ Predictor:
|
|
| 15 |
distance: L1
|
| 16 |
overlap: 2
|
| 17 |
mixed_precision: false
|
| 18 |
-
grid_spacing:
|
| 19 |
-
displacement_half_width:
|
| 20 |
-
iterations:
|
| 21 |
-
learning_rate:
|
| 22 |
-
regularization_weight: 1.
|
| 23 |
-
grid_shrink:
|
|
|
|
| 24 |
subset_features: []
|
| 25 |
linear: true
|
| 26 |
linear_iterations: 200
|
|
|
|
| 15 |
distance: L1
|
| 16 |
overlap: 2
|
| 17 |
mixed_precision: false
|
| 18 |
+
grid_spacing: 6
|
| 19 |
+
displacement_half_width: 4
|
| 20 |
+
iterations: 80
|
| 21 |
+
learning_rate: 1.0
|
| 22 |
+
regularization_weight: 1.25
|
| 23 |
+
grid_shrink: 2
|
| 24 |
+
control_grid_smoothing: 3
|
| 25 |
subset_features: []
|
| 26 |
linear: true
|
| 27 |
linear_iterations: 200
|
|
@@ -15,12 +15,13 @@ Predictor:
|
|
| 15 |
distance: L1
|
| 16 |
overlap: 2
|
| 17 |
mixed_precision: false
|
| 18 |
-
grid_spacing:
|
| 19 |
-
displacement_half_width:
|
| 20 |
-
iterations:
|
| 21 |
-
learning_rate:
|
| 22 |
-
regularization_weight: 1.
|
| 23 |
-
grid_shrink:
|
|
|
|
| 24 |
subset_features: []
|
| 25 |
linear: true
|
| 26 |
linear_iterations: 200
|
|
|
|
| 15 |
distance: L1
|
| 16 |
overlap: 2
|
| 17 |
mixed_precision: false
|
| 18 |
+
grid_spacing: 6
|
| 19 |
+
displacement_half_width: 4
|
| 20 |
+
iterations: 80
|
| 21 |
+
learning_rate: 1.0
|
| 22 |
+
regularization_weight: 1.25
|
| 23 |
+
grid_shrink: 2
|
| 24 |
+
control_grid_smoothing: 3
|
| 25 |
subset_features: []
|
| 26 |
linear: true
|
| 27 |
linear_iterations: 200
|