VBoussot commited on
Commit
bd726b5
·
verified ·
1 Parent(s): 90b796d

fix: the ConvexAdam presets run the settings ConvexAdam runs

Browse files

The 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.

ConvexAdam_Coarse/Prediction.yml CHANGED
@@ -15,12 +15,13 @@ Predictor:
15
  distance: L1
16
  overlap: 2
17
  mixed_precision: false
18
- grid_spacing: 4
19
- displacement_half_width: 6
20
- iterations: 150
21
- learning_rate: 0.2
22
- regularization_weight: 1.0
23
- grid_shrink: 4
 
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
ConvexAdam_Composite/Prediction.yml CHANGED
@@ -15,12 +15,13 @@ Predictor:
15
  distance: L1
16
  overlap: 2
17
  mixed_precision: false
18
- grid_spacing: 4
19
- displacement_half_width: 6
20
- iterations: 150
21
- learning_rate: 0.2
22
- regularization_weight: 1.0
23
- grid_shrink: 4
 
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