Pith. sign in

REVIEW 4 major objections 5 minor 94 references

Improving Generalization of Medical Image Registration Foundation Model

T0 review · 4 major / 5 minor · reviewed 2026-08-15 · deepseek-v4-flash

Pith's one-line read Training the uniGradICON medical image registration foundation model with Sharpness-Aware Minimization improves cross-dataset registration accuracy.

desk verdict Plausible SAM application to a registration foundation model, but the central 'significant improvement' claim rests on a boxplot with no statistics and a table that omits the primary baseline. read the letter →

arxiv 2505.06527 v1 pith:EZN4TKQV submitted 2025-05-10 cs.CV cs.AI

classification cs.CVcs.AI
keywords Sharpness-AwareMinimization(SAM)medicalimageregistrationfoundationmodelgeneralizationcross-datasetdeformableuniGradICONlosslandscape
verification ladder T0 review T1 audit T2 compute T3 formal

The pith

A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.

The reading

This paper addresses a known weakness of medical image registration foundation models: they lose accuracy when applied to datasets, anatomies, or modalities they were not trained on. The authors propose training the uniGradICON foundation model with Sharpness-Aware Minimization (SAM), an optimizer that steers the model toward flat regions of the loss landscape, as a way to improve cross-dataset generalization. Their experiments on five held-out datasets — cardiac, liver, abdominal MR-CT, hippocampus, and lung CT — show that the SAM-trained model achieves consistently higher Dice overlap scores than the same model without SAM. The paper argues that this flat-minima training makes the model more robust across different anatomical structures and imaging conditions, and it visualizes a flatter loss landscape for the SAM-trained model in support.

What carries the argument

Sharpness-Aware Minimization (SAM) is the load-bearing mechanism: an optimization scheme that minimizes, not the training loss at the current parameters, but the worst-case loss in a small ball around them, by perturbing weights along the gradient direction by a radius ρ before computing each update. The paper applies SAM on top of the uniGradICON foundation model, whose architecture is built from ICON's downsampling (Down) and two-step (TS) operators and trained with the GradICON loss — a localized normalized cross-correlation similarity term plus a gradient inverse-consistency regularizer. SAM's adversarial perturbation is computed with the efficient first-order approximation of the inner maximization, so the extra cost per update is roughly one additional forward-backward pass. The flatness of the resulting loss landscape is the explanatory link the paper draws to generalization.

What would settle it

Retrain the baseline uniGradICON model from scratch on the same composite training dataset, using the same random seed, number of epochs, learning rate, and loss weights, with SAM as the only added component, and measure Dice overlap on the same five test datasets. If the Dice differences between the two models disappear or become statistically insignificant, the paper's central claim would be falsified.

Watch

Extended reading notes

Core claim

The central claim is that incorporating Sharpness-Aware Minimization into the training of the uniGradICON foundation model improves its generalization across medical image registration tasks. SAM replaces the ordinary gradient update with a two-step procedure: first compute the gradient at the current weights, take a small step of size ρ in that gradient direction to find a perturbation that maximizes the training loss (an adversarial point within a neighborhood), then compute the actual update gradient at that perturbed point. This minimax objective biases training toward minima whose neighborhoods are uniformly low in loss — flat minima — which the paper finds produces a flatter loss landscape than the base model. The paper evaluates the approach on the ACDC cardiac and SLIVER liver datasets against four comparison baselines and against uniGradICON on five datasets, reporting that the SAM-trained model achieves the highest average Dice scores and near-zero or zero percentages of folded voxels.

Load-bearing premise

The claim that SAM alone causes the improvement rests on the assumption that the comparison uniGradICON model was trained under identical conditions, with the same data sampling, epochs, loss weights, and random seeds, so that adding SAM is the only change.

Editorial extensions

If this is right

  • The SAM-trained model reports higher average Dice scores than all compared baselines on the ACDC and SLIVER datasets, with folded-voxel percentages near or exactly zero.
  • Across the five ablation datasets, the method consistently achieves significantly higher Dice coefficients than uniGradICON, indicating stronger cross-dataset generalization.
  • The loss-landscape visualizations are flatter for the SAM-trained model, matching the proposed mechanism that flat minima generalize better.
  • Because SAM adds only one hyperparameter and is compatible with any first-order optimizer, the same recipe can be applied to other registration networks without architectural changes.

Reading between the lines

Editorial extensions of the paper, not claims the author makes directly.

  • A controlled re-training study with identical seeds and hyperparameters, varying only SAM, would isolate the optimizer's true contribution from any training-schedule differences in the published comparison.
  • The method's success on both MRI and CT across four anatomical regions suggests the flat-minima recipe is modality-agnostic and might transfer to 2D imaging (X-ray, ultrasound) with minimal adjustment.
  • Quantifying the sharpness of the found minimum directly — for instance, by measuring the curvature of the loss surface — rather than relying on visual loss-landscape plots would let future work verify the generalization mechanism more rigorously.
Share X Bluesky LinkedIn Reddit HN

Signed reviews

No signed human review yet.

Editorial analysis

A structured set of objections, weighed in public.

Desk editor's note, referee report, and a circularity audit.

Referee Report

4 major / 5 minor

Summary. The paper proposes incorporating Sharpness-Aware Minimization (SAM) into the uniGradICON medical image registration foundation model. The method replaces the base optimizer's gradient with one computed at an adversarial perturbation, supposedly leading to flatter minima and improved cross-dataset generalization. The authors train on a composite dataset of four sources (COPDGene, OAI, HCP, L2R-Abdomen) and evaluate on five test sets (ACDC, SLIVER, and three Learn2Reg tasks). The central claim, stated in Section IV-D, is that adding SAM to uniGradICON 'consistently achieves significantly higher Dice coefficients on all five datasets,' supported by a boxplot (Fig. 5) and a qualitative loss-landscape visualization (Fig. 6). Quantitative comparisons against other methods on ACDC and SLIVER are given in Table II.

Significance. If substantiated, the result would be of practical interest: a drop-in modification (SAM) to an existing foundation model that improves cross-dataset generalization with no architectural change is a useful contribution. The paper also makes the method reproducible by releasing code. However, the core evidence for the central claim is currently descriptive rather than statistical. The reported margins over the best runner-up in Table II are small (0.0038 on ACDC, 0.0042 on SLIVER), and the primary baseline uniGradICON is absent from that table. The paper's strength is its clear framing of a plausible hypothesis (flat minima improve registration generalization) and the use of established components (GradICON/uniGradICON, SAM). The weakness is that the direct head-to-head comparison is not quantitatively reported, so the central claim is not yet established.

major comments (4)
  1. [Section IV-D, Fig. 5] The claim that 'our method consistently achieves significantly higher Dice coefficients on all five datasets' is supported only by a boxplot. The figure reports no means, standard deviations, per-dataset sample sizes, or statistical tests. The word 'significantly' is therefore unsupported. Please provide the underlying numerical results (e.g., mean ± std for each dataset and each method) and, where appropriate, a paired significance test or confidence intervals, to substantiate the claim.
  2. [Section IV-C, Table II] The primary baseline uniGradICON is missing from Table II, the only quantitative comparison table. Without numbers for uniGradICON on ACDC and SLIVER, the reader cannot assess whether SAM improves over its base model in the two datasets that do have quantitative results. Please add uniGradICON results to Table II, or clearly state why they are omitted.
  3. [Section IV-D, Fig. 5] The paper does not state whether the uniGradICON baseline in Fig. 5 was retrained from scratch on the identical composite dataset with the same 800/200 epochs, loss weights (λ=1.5), learning rate (η=5e-5), random seeds, and sampling protocol, with SAM as the only difference. If the baseline numbers come from a published checkpoint or a different training procedure, the claimed generalization gain could be caused by training details rather than by SAM. Please specify the exact training setup for both models.
  4. [Section IV-D, Fig. 6] The loss-landscape visualization is qualitative and does not quantify flatness (e.g., via the trace of the Hessian, the largest eigenvalue, or a sharpness measure). As presented, Fig. 6 is suggestive but does not independently establish the mechanism of improved generalization. Please either add a quantitative flatness measure or temper the mechanistic claim.
minor comments (5)
  1. [Abstract] The abstract contains a malformed URL: 'https://github.com/Promise13/fm_sam}{https://github.com/Promise13/fm\_sam'. Please correct the formatting.
  2. [Section III-B] The section heading 'Taining loss' should be 'Training loss'.
  3. [Section II-D] The phrase 'this method has effectively improved model performance' is vague; please specify the tasks and the magnitude of improvements reported in prior work.
  4. [Section IV-A, Table I] The 'SLIVER' dataset is listed as MRI, but the cited SLIVER dataset (Heimann et al., 2009) is a CT dataset. Please verify the modality label.
  5. [Section IV-C] The boxplot in Fig. 3 is described in the text, but no caption appears in the manuscript body. Please add a complete caption for Fig. 3.

Circularity Check

0 steps flagged · score 0.0 of 10

No circularity: the core contribution is an empirical training modification (SAM on uniGradICON), evaluated against external baselines and the same base model; no derivation reduces to its own inputs.

full rationale

The paper's method is externally sourced and independently applied: the SAM objective (Eqs. 5-7) is taken from Foret et al. [64], the registration network and loss (Eq. 8) come from GradICON/uniGradICON [60,82], and the hyperparameters (rho=0.1, lambda=1.5, learning rate, epochs) are stated as fixed settings rather than fitted to the test results. The central comparison in Table II is against external baselines (Elastix, VoxelMorph, DiffuseMorph, TransMorph) on held-out ACDC and SLIVER data, and the ablation in Fig. 5 compares the SAM-trained model against the uniGradICON base model. No test-set values are folded back into the method, no parameter is defined in terms of the target metric, and no load-bearing argument reduces to a self-citation. The paper does have reporting weaknesses: the Fig. 5 ablation shows only boxplots without numbers or significance tests, and the text does not explicitly state that the uniGradICON baseline was retrained under identical conditions with SAM as the only difference. Those are evidence-quality and correctness-risk issues, not circularity: a weak empirical comparison is still an empirical comparison. Self-citations in the reference list concern prior medical-image-registration and deepfake/segmentation applications and are not used to justify the SAM-generalization claim. Accordingly, no circular step can be exhibited, and the appropriate score is 0.

Assumptions & free parameters 3 free parameters · 4 assumptions · 0 invented entities

The paper introduces no new entities, forces, or conserved quantities. Its free parameters are standard training hyperparameters with values inherited from prior work. The main non-standard assumption is that the uniGradICON baseline is trained identically except for SAM, which is not demonstrated.

free parameters (3)
  • sam_neighborhood_size_rho = 0.1
    SAM perturbation radius, set by hand following prior SAM practice; not tuned to the test sets.
  • inverse_consistency_weight_lambda = 1.5
    Weight for the GradICON inverse-consistency regularizer, taken from GradICON defaults.
  • learning_rate_eta = 5e-5
    Adam learning rate, taken from GradICON defaults; not fit to the data.
assumptions (4)
  • standard math SAM's minimax objective and first-order approximation (Eq. 5-7) are valid and improve flatness.
    Taken from Foret et al. [64]; the paper applies it without modification.
  • domain assumption Flat minima generalize better to new datasets.
    Imported from SAM literature; stated in Section II-D and used to interpret Fig. 6.
  • domain assumption The composite training set (COPDGene, OAI, HCP, L2R-Abdomen) is a sufficient basis for a registration foundation model.
    Same training set as uniGradICON [60]; no analysis of domain shift inside the training data.
  • ad hoc to paper The comparison baseline uniGradICON is trained under identical conditions except for SAM.
    Section IV-D presents the comparison without reporting baseline training details.

how reviews work

0 comments
Cite this review

Pith. "Pith review of Improving Generalization of Medical Image Registration Foundation Model." pith.science (2026). https://pith.science/paper/EZN4TKQV

@misc{pith2026250506527,
  author       = {Pith},
  title        = {Pith review of: Improving Generalization of Medical Image Registration Foundation Model},
  year         = {2026},
  howpublished = {\url{https://pith.science/paper/EZN4TKQV}},
  note         = {Machine review of arXiv:2505.06527}
}
read the original abstract

Deformable registration is a fundamental task in medical image processing, aiming to achieve precise alignment by establishing nonlinear correspondences between images. Traditional methods offer good adaptability and interpretability but are limited by computational efficiency. Although deep learning approaches have significantly improved registration speed and accuracy, they often lack flexibility and generalizability across different datasets and tasks. In recent years, foundation models have emerged as a promising direction, leveraging large and diverse datasets to learn universal features and transformation patterns for image registration, thus demonstrating strong cross-task transferability. However, these models still face challenges in generalization and robustness when encountering novel anatomical structures, varying imaging conditions, or unseen modalities. To address these limitations, this paper incorporates Sharpness-Aware Minimization (SAM) into foundation models to enhance their generalization and robustness in medical image registration. By optimizing the flatness of the loss landscape, SAM improves model stability across diverse data distributions and strengthens its ability to handle complex clinical scenarios. Experimental results show that foundation models integrated with SAM achieve significant improvements in cross-dataset registration performance, offering new insights for the advancement of medical image registration technology. Our code is available at https://github.com/Promise13/fm_sam}{https://github.com/Promise13/fm\_sam.

Figures

Figures reproduced from arXiv: 2505.06527 by the authors.

Figure 2
Figure 2. Schematic of the SAM parameter update. Algorithm 1 SAM Algorithm. Input: Training set S △ = Un i=1 {(xi , yi)} ,loss function L : W × X × Y −→ R+ ,batch size b,step size η > 0, neighborhood size ρ > 0 (default 0.15) Output: Model trained with SAM Initialize weights w0, t = 0; while not converged do Sample batch B = {(x1, y1, ...(xb, yb))}; Compute gradient▽wLB (w) of the batch’s training loss; Compute ϵˆ(w) per equa… view at source ↗
Figure 4
Figure 4. An example of axial slices for cardiac MR image registration, [PITH_FULL_IMAGE:figures/full_fig_p005_4.png] view at source ↗
Figure 6
Figure 6. Loss landscape visualization. Left: uniGradICON. Right: Ours. [PITH_FULL_IMAGE:figures/full_fig_p006_6.png] view at source ↗

Discussion (0). Continue with ORCID to comment.

Reference graph

Works this paper leans on

94 extracted references · 65 canonical work pages

  1. [1]

    A fast diffeomorphic image registration algo- rithm,

    J. Ashburner, “A fast diffeomorphic image registration algo- rithm,” Neuroimage, vol. 38, no. 1, pp. 95–113, 2007

  2. [2]

    Patch-based discrete registration of clinical brain images,

    A. V . Dalca, A. Bobu, N. S. Rost, and P. Golland, “Patch-based discrete registration of clinical brain images,” in Patch-Based Techniques in Medical Imaging . Springer, 2016

  3. [3]

    Dense image registration through mrfs and efficient linear programming,

    B. Glocker et al. , “Dense image registration through mrfs and efficient linear programming,” MIA, 2008

  4. [4]

    Image matching as a diffusion process: an analogy with maxwell’s demons,

    J.-P. Thirion, “Image matching as a diffusion process: an analogy with maxwell’s demons,” MIA, 1998

  5. [5]

    Image registration methods: a survey,

    B. Zitova and J. Flusser, “Image registration methods: a survey,” Image and vision computing , 2003

  6. [6]

    A survey of medical image registration,

    J. A. Maintz and M. A. Viergever, “A survey of medical image registration,” Medical image analysis , 1998

  7. [7]

    Attention guided policy optimization for 3d medical image registration,

    J. Hu, Z. Shuai, X. Wang, S. Hu, S. Sun, S. Lyu, and X. Wu, “Attention guided policy optimization for 3d medical image registration,” IEEE Access , vol. 11, pp. 65 546–65 558, 2023

  8. [8]

    Stochastic planner-actor-critic for unsu- pervised deformable image registration,

    Z. Luo, J. Hu, X. Wang, S. Hu, B. Kong, Y . Yin, Q. Song, X. Wu, and S. Lyu, “Stochastic planner-actor-critic for unsu- pervised deformable image registration,” in AAAI, 2022

Show all 94 references
  1. [9]

    Fast parametric elastic image registra- tion,

    J. Kybic and M. Unser, “Fast parametric elastic image registra- tion,” TIP, 2003

  2. [10]

    Multiresolution elastic matching,

    R. Bajcsy et al. , “Multiresolution elastic matching,” Computer vision, graphics, and image processing , 1989

  3. [11]

    Hammer: hierarchical attribute matching mech- anism for elastic registration,

    D. Shen et al., “Hammer: hierarchical attribute matching mech- anism for elastic registration,” TMI, 2002

  4. [12]

    A simple and robust feature point matching algorithm based on restricted spatial order constraints for aerial image registration,

    Z. Liu et al. , “A simple and robust feature point matching algorithm based on restricted spatial order constraints for aerial image registration,” IEEE Transactions on Geoscience and Remote Sensing , 2011

  5. [13]

    New feature points based on geometric invariants for 3d image registration,

    J.-P. Thirion, “New feature points based on geometric invariants for 3d image registration,” IJCV, 1996

  6. [14]

    Elastix: a toolbox for intensity-based medical image registration,

    S. Klein et al. , “Elastix: a toolbox for intensity-based medical image registration,” TMI, 2009

  7. [15]

    The adaptive bases algorithm for intensity- based nonrigid image registration,

    G. K. Rohde et al., “The adaptive bases algorithm for intensity- based nonrigid image registration,” TMI, 2003

  8. [16]

    Fast free-form deformation using graph- ics processing units,

    M. Modat et al. , “Fast free-form deformation using graph- ics processing units,” Computer methods and programs in biomedicine, 2010

  9. [17]

    Global image registration using a symmetric block- matching approach,

    ——, “Global image registration using a symmetric block- matching approach,” Journal of medical imaging , 2014

  10. [18]

    Deep learning,

    Y . LeCun, Y . Bengio, and G. Hinton, “Deep learning,” nature, vol. 521, no. 7553, pp. 436–444, 2015

  11. [19]

    Goodfellow, Y

    I. Goodfellow, Y . Bengio, A. Courville, and Y . Bengio, Deep learning. MIT press Cambridge, 2016, vol. 1, no. 2

  12. [20]

    Diffusion-empowered autoprompt medsam,

    P. Huang, S. Hu, B. Peng, J. Zhang, H. Zhu, X. Wu, and X. Wang, “Diffusion-empowered autoprompt medsam,” arXiv preprint arXiv:2502.06817, 2025

  13. [21]

    Graph fairness via authen- tic counterfactuals: Tackling structural and causal challenges,

    Z. Wang, Z. Yin, Y . Zhang, L. Yang, T. Zhang, N. Pissinou, Y . Cai, S. Hu, Y . Li, L. Zhaoet al., “Graph fairness via authen- tic counterfactuals: Tackling structural and causal challenges,” ACM SIGKDD Explorations Newsletter , 2025

  14. [22]

    Fg-smote: Towards fair node classification with graph neural network,

    ——, “Fg-smote: Towards fair node classification with graph neural network,” ACM SIGKDD Explorations Newsletter , vol. 26, no. 2, pp. 99–108, 2025

  15. [23]

    Improving generalization for ai-synthesized voice detection,

    H. Ren, L. Li, C.-H. Liu, X. Wang, and S. Hu, “Improving generalization for ai-synthesized voice detection,” AAAI, 2024

  16. [24]

    Ai-face: A million-scale de- mographically annotated ai-generated face dataset and fairness benchmark,

    L. Lin, X. Wang, S. Hu et al. , “Ai-face: A million-scale de- mographically annotated ai-generated face dataset and fairness benchmark,” CVPR, 2025

  17. [25]

    Vb-kgn: Variational bayesian kernel generation networks for motion image deblurring,

    Y . Fu, X. Zhu, X. Li, X. Wang, X. Wu, S. Hu, Y . Wu, S. Lyu, and W. Liu, “Vb-kgn: Variational bayesian kernel generation networks for motion image deblurring,” TMM, 2024

  18. [26]

    A self-learning multimodal approach for fake news detection,

    H. Chen, H. Guo, B. Hu, S. Hu, J. Hu, S. Lyu, X. Wu, and X. Wang, “A self-learning multimodal approach for fake news detection,” arXiv, 2024

  19. [27]

    U-medsam: Uncertainty-aware medsam for medical image segmentation,

    X. Wang, X. Liu, P. Huang, P. Huang, S. Hu, and H. Zhu, “U-medsam: Uncertainty-aware medsam for medical image segmentation,” arXiv preprint arXiv:2408.08881 , 2024

  20. [28]

    Repmedgraf: Re-parameterization medical gener- ated radiation field for improved 3d image reconstruction,

    R. Sun, F. Liao, J. He, Y . Liu, W. Yi, Q. Fan, S. Hu, X. Wang, and J. Hu, “Repmedgraf: Re-parameterization medical gener- ated radiation field for improved 3d image reconstruction,” in A VSS. IEEE, 2024

  21. [29]

    Cgd-net: A hybrid end-to-end network with gating decoding for liver tumor segmentation from ct images,

    X. Zhu, T. Liu, Z. Liu, O. Shaobo, X. Wang, S. Hu, and F. Ding, “Cgd-net: A hybrid end-to-end network with gating decoding for liver tumor segmentation from ct images,” in A VSS. IEEE, 2024

  22. [30]

    Qrnn- transformer: Recognizing textual entailment,

    X. Zhu, Z. Yan, W. Wang, S. Hu, X. Wang, and C. Liu, “Qrnn- transformer: Recognizing textual entailment,” in A VSS. IEEE, 2024

  23. [31]

    Enhancing vehicle re-identification and matching for weaving analysis,

    M. Qiu, W. Lin, S. Chien, L. Christopher, Y . Chen, and S. Hu, “Enhancing vehicle re-identification and matching for weaving analysis,” in A VSS. IEEE, 2024

  24. [32]

    Uncertainty-aware explainable recommendation with large language models,

    Y . Peng, H. Chen, C.-S. Lin, G. Huang, J. Hu, H. Guo, B. Kong, S. Hu, X. Wu, and X. Wang, “Uncertainty-aware explainable recommendation with large language models,” in IJCNN. IEEE, 2024, pp. 1–8

  25. [33]

    Llm-medqa: Enhancing medical question answering through case studies in large language models,

    H. Yang, H. Chen, H. Guo, Y . Chen, C.-S. Lin, S. Hu, J. Hu, X. Wu, and X. Wang, “Llm-medqa: Enhancing medical question answering through case studies in large language models,” IJCNN, 2025

  26. [34]

    Real-time lane-wise traffic monitoring in optimal rois,

    M. Qiu, W. Lin, L. A. Christopher, S. Chien, Y . Chen, and S. Hu, “Real-time lane-wise traffic monitoring in optimal rois,” MIPR, 2024

  27. [35]

    Masked conditional diffusion model for enhancing deepfake detection,

    T. Chen, S. Yang, S. Hu, Z. Fang, Y . Fu, X. Wu, and X. Wang, “Masked conditional diffusion model for enhancing deepfake detection,” arXiv preprint arXiv:2402.00541 , 2024

  28. [36]

    Detecting multi- media generated by large ai models: A survey,

    L. Lin, N. Gupta, Y . Zhang, H. Ren, C.-H. Liu, F. Ding, X. Wang, X. Li, L. Verdoliva, and S. Hu, “Detecting multi- media generated by large ai models: A survey,” arXiv preprint arXiv:2402.00045, 2024

  29. [37]

    Efficient image super-resolution via symmetric visual attention network,

    Q. Fan, C. Wu, S. Hu, X. Wu, X. Wang, and J. Hu, “Efficient image super-resolution via symmetric visual attention network,” in IJCNN. IEEE, 2024

  30. [38]

    Fairness in survival analysis with dis- tributionally robust optimization,

    S. Hu and G. H. Chen, “Fairness in survival analysis with dis- tributionally robust optimization,” Journal of Machine Learning Research, vol. 25, no. 246, pp. 1–85, 2024

  31. [39]

    Synthesizing black-box anti- forensics deepfakes with high visual quality,

    B. Fan, S. Hu, and F. Ding, “Synthesizing black-box anti- forensics deepfakes with high visual quality,” inICASSP. IEEE, 2024

  32. [40]

    Umednerf: Uncertainty-aware single view volumetric rendering for medical neural radiance fields,

    J. Hu, Q. Fan, S. Hu, S. Lyu, X. Wu, and X. Wang, “Umednerf: Uncertainty-aware single view volumetric rendering for medical neural radiance fields,” in ISBI. IEEE, 2024, pp. 1–4

  33. [41]

    X-transfer: A transfer learning-based framework for gan-generated fake image detection,

    L. Zhang, H. Chen, S. Hu, B. Zhu, C.-S. Lin, X. Wu, J. Hu, and X. Wang, “X-transfer: A transfer learning-based framework for gan-generated fake image detection,” in IJCNN. IEEE, 2024

  34. [42]

    Rmbench: Benchmarking deep reinforcement learning for robotic manipulator control,

    Y . Xiang, X. Wang, S. Hu, B. Zhu, X. Huang, X. Wu, and S. Lyu, “Rmbench: Benchmarking deep reinforcement learning for robotic manipulator control,” in IROS. IEEE, 2023

  35. [43]

    Improving cross-dataset deepfake detection with deep infor- mation decomposition,

    S. Yang, S. Hu, B. Zhu, Y . Fu, S. Lyu, X. Wu, and X. Wang, “Improving cross-dataset deepfake detection with deep infor- mation decomposition,” arXiv, 2023

  36. [44]

    Deep reinforcement learning for image-to-image translation,

    X. Wang, Z. Luo, J. Hu, C. Feng, S. Hu, B. Zhu, X. Wu, and S. Lyu, “Deep reinforcement learning for image-to-image translation,” arXiv preprint arXiv:2309.13672 , 2023

  37. [45]

    Outlier robust adversarial training,

    S. Hu, Z. Yang, X. Wang, Y . Ying, and S. Lyu, “Outlier robust adversarial training,” in Asian Conference on Machine Learning. PMLR, 2024, pp. 454–469

  38. [46]

    Attacking identity semantics in deepfakes via deep feature fusion,

    B. Fan, Z. Jiang, S. Hu, and F. Ding, “Attacking identity semantics in deepfakes via deep feature fusion,” in MIPR. IEEE, 2023

  39. [47]

    Improving fairness in deepfake detection,

    Y . Ju, S. Hu, S. Jia, G. H. Chen, and S. Lyu, “Improving fairness in deepfake detection,” in WACV, 2024

  40. [48]

    Attention-driven tree-structured convolutional lstm for high dimensional data understanding,

    Y . Lu, B. Kong, F. Gao, K. Cao, S. Lyu, S. Zhang, S. Hu, Y . Yin, and X. Wang, “Attention-driven tree-structured convolutional lstm for high dimensional data understanding,” Frontiers in Physics, vol. 11, p. 1095277, 2023

  41. [49]

    Attacking important pixels for anchor-free detectors,

    Y . Xie, S. Hu, X. Wang, Q. Liao, B. Zhu, X. Wu, and S. Lyu, “Attacking important pixels for anchor-free detectors,” arXiv preprint arXiv:2301.11457, 2023

  42. [50]

    Ntire 2023 challenge on efficient super-resolution: Methods and results,

    Y . Li, Y . Zhang, R. Timofte, L. Van Gool, L. Yu, Y . Li, X. Li, T. Jiang, Q. Wu, M. Han et al. , “Ntire 2023 challenge on efficient super-resolution: Methods and results,” in CVPR, 2023

  43. [51]

    Harnessing the power of text-image contrastive models for automatic detection of online misinformation,

    H. Chen, P. Zheng, X. Wang, S. Hu, B. Zhu, J. Hu, X. Wu, and S. Lyu, “Harnessing the power of text-image contrastive models for automatic detection of online misinformation,” in CVPR, 2023

  44. [52]

    Rank-based decomposable losses in machine learning: A survey,

    S. Hu, X. Wang, and S. Lyu, “Rank-based decomposable losses in machine learning: A survey,” IEEE Transactions on Pattern Analysis and Machine Intelligence , 2023

  45. [53]

    Distributionally robust survival anal- ysis: A novel fairness loss without demographics,

    S. Hu and G. H. Chen, “Distributionally robust survival anal- ysis: A novel fairness loss without demographics,” in Machine Learning for Health . PMLR, 2022, pp. 62–87

  46. [54]

    Deformable image registration based on similarity- steered cnn regression,

    X. Cao, J. Yang, J. Zhang, D. Nie, M. Kim, Q. Wang, and D. Shen, “Deformable image registration based on similarity- steered cnn regression,” in MICCAI. Springer, 2017

  47. [55]

    Fast predictive image registration—a deep learning approach., 2017, 158,

    X. Yang, R. Kwitt, M. Styner, and M. N. Quicksilver, “Fast predictive image registration—a deep learning approach., 2017, 158,” DOI: https://doi. org/10.1016/j. neuroimage , vol. 8, pp. 378–396, 2017

  48. [56]

    V oxelmorph: a learning framework for deformable medical image registration,

    G. Balakrishnan et al. , “V oxelmorph: a learning framework for deformable medical image registration,” TMI, 2019

  49. [57]

    Nonrigid image registration using multi-scale 3d convolutional neural networks,

    H. Sokooti et al., “Nonrigid image registration using multi-scale 3d convolutional neural networks,” in MICCAI. Springer, 2017

  50. [58]

    A deep learning framework for unsuper- vised affine and deformable image registration,

    B. D. De V os et al. , “A deep learning framework for unsuper- vised affine and deformable image registration,” MIA, 2019

  51. [59]

    Svf-net: learning deformable image regis- tration using shape matching,

    M.-M. Roh ´e et al. , “Svf-net: learning deformable image regis- tration using shape matching,” in MICCAI. Springer, 2017

  52. [60]

    unigradicon: A foundation model for medical image registration,

    L. Tian et al. , “unigradicon: A foundation model for medical image registration,” in MICCAI. Springer, 2024

  53. [61]

    Advancing medical image registration with the vision foundation model (vfm): A modular pre-trained framework,

    H. Wang, Q. Zhou, and Z. Pan, “Advancing medical image registration with the vision foundation model (vfm): A modular pre-trained framework,” in 2024 7th International Conference on Information Communication and Signal Processing (ICI- CSP). IEEE, 2024, pp. 1084–1088

  54. [62]

    Multigradicon: A foundation model for multimodal medical image registration,

    B. Demir, L. Tian, H. Greer, R. Kwitt, F.-X. Vialard, R. S. J. Est´epar, S. Bouix, R. Rushmore, E. Ebrahim, and M. Ni- ethammer, “Multigradicon: A foundation model for multimodal medical image registration,” in International Workshop on Biomedical Image Registration . Springer...

  55. [63]

    On the challenges and perspectives of foundation models for medical image analysis,

    S. Zhang and D. Metaxas, “On the challenges and perspectives of foundation models for medical image analysis,” Medical image analysis , vol. 91, p. 102996, 2024

  56. [64]

    Sharpness- aware minimization for efficiently improving generalization,

    P. Foret, A. Kleiner, H. Mobahi, and B. Neyshabur, “Sharpness- aware minimization for efficiently improving generalization,” arXiv preprint arXiv:2010.01412 , 2020

  57. [65]

    Robust light- weight facial affective behavior recognition with clip,

    L. Lin, S. Papabathini, X. Wang, and S. Hu, “Robust light- weight facial affective behavior recognition with clip,” MIPR, 2024

  58. [66]

    Robust clip-based detector for exposing diffusion model-generated images,

    L. Lin, I. Amerini, X. Wang, S. Hu et al. , “Robust clip-based detector for exposing diffusion model-generated images,”MIPR, 2024

  59. [67]

    Preserving fairness generalization in deepfake detection,

    L. Lin, X. He, Y . Ju, X. Wang, F. Ding, and S. Hu, “Preserving fairness generalization in deepfake detection,” in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2024, pp. 16 815–16 825

  60. [68]

    Spatial transformation and registration of brain images using elastically deformable models,

    C. Davatzikos, “Spatial transformation and registration of brain images using elastically deformable models,” Computer Vision and Image Understanding , vol. 66, no. 2, pp. 207–222, 1997

  61. [69]

    Nonrigid registration using free- form deformations: application to breast mr images,

    D. Rueckert, L. I. Sonoda, C. Hayes, D. L. Hill, M. O. Leach, and D. J. Hawkes, “Nonrigid registration using free- form deformations: application to breast mr images,” IEEE transactions on medical imaging , vol. 18, no. 8, pp. 712–721, 1999

  62. [70]

    Computing large deformation metric mappings via geodesic flows of diffeo- morphisms,

    M. F. Beg, M. I. Miller, A. Trouv ´e, and L. Younes, “Computing large deformation metric mappings via geodesic flows of diffeo- morphisms,” International journal of computer vision , vol. 61, pp. 139–157, 2005

  63. [71]

    Frequency diffeomorphisms for efficient image registration,

    M. Zhang, R. Liao, A. V . Dalca, E. A. Turk, J. Luo, P. E. Grant, and P. Golland, “Frequency diffeomorphisms for efficient image registration,” in Information Processing in Medical Imaging: 25th International Conference, IPMI 2017, Boone, NC, USA, June 25-30, 2017, Proceedings...

  64. [72]

    Diffeo- morphic demons: Efficient non-parametric image registration,

    T. Vercauteren, X. Pennec, A. Perchant, and N. Ayache, “Diffeo- morphic demons: Efficient non-parametric image registration,” NeuroImage, vol. 45, no. 1, pp. S61–S72, 2009

  65. [73]

    Sym- metric diffeomorphic image registration with cross-correlation: evaluating automated labeling of elderly and neurodegenerative brain,

    B. B. Avants, C. L. Epstein, M. Grossman, and J. C. Gee, “Sym- metric diffeomorphic image registration with cross-correlation: evaluating automated labeling of elderly and neurodegenerative brain,” Medical image analysis , vol. 12, no. 1, pp. 26–41, 2008

  66. [74]

    Recent advances in convolutional neural networks,

    J. Gu, Z. Wang, J. Kuen, L. Ma, A. Shahroudy, B. Shuai, T. Liu, X. Wang, G. Wang, J. Cai et al. , “Recent advances in convolutional neural networks,” Pattern recognition, vol. 77, pp. 354–377, 2018

  67. [75]

    Large deformation diffeomorphic image registration with laplacian pyramid networks,

    T. C. Mok and A. C. Chung, “Large deformation diffeomorphic image registration with laplacian pyramid networks,” in Medical Image Computing and Computer Assisted Intervention–MICCAI 2020: 23rd International Conference, Lima, Peru, October 4–8, 2020, Proceedings, Part III 23 . ...

  68. [76]

    Cyclemorph: cycle consistent unsupervised deformable image registration,

    B. Kim, D. H. Kim, S. H. Park, J. Kim, J.-G. Lee, and J. C. Ye, “Cyclemorph: cycle consistent unsupervised deformable image registration,” Medical image analysis , vol. 71, p. 102036, 2021

  69. [77]

    Generative adversarial networks,

    I. Goodfellow, J. Pouget-Abadie, M. Mirza, B. Xu, D. Warde- Farley, S. Ozair, A. Courville, and Y . Bengio, “Generative adversarial networks,” Communications of the ACM , vol. 63, no. 11, pp. 139–144, 2020

  70. [78]

    Diffusemorph: Unsupervised deformable image registration using diffusion model,

    B. Kim, I. Han, and J. C. Ye, “Diffusemorph: Unsupervised deformable image registration using diffusion model,” in Eu- ropean conference on computer vision . Springer, 2022, pp. 347–364

  71. [79]

    Denoising diffusion probabilistic models,

    J. Ho, A. Jain, and P. Abbeel, “Denoising diffusion probabilistic models,” Advances in neural information processing systems , vol. 33, pp. 6840–6851, 2020

  72. [80]

    Attention is all you need,

    A. Vaswani, N. Shazeer, N. Parmar, J. Uszkoreit, L. Jones, A. N. Gomez, Ł. Kaiser, and I. Polosukhin, “Attention is all you need,” Advances in neural information processing systems , vol. 30, 2017

  73. [81]

    Transmorph: Transformer for unsupervised medical image registration,

    J. Chen, E. C. Frey, Y . He, W. P. Segars, Y . Li, and Y . Du, “Transmorph: Transformer for unsupervised medical image registration,” Medical image analysis , vol. 82, p. 102615, 2022

  74. [82]

    Gradicon: Approximate diffeomorphisms via gradient inverse consistency,

    L. Tian, H. Greer, F.-X. Vialard, R. Kwitt, R. S. J. Est ´epar, R. J. Rushmore, N. Makris, S. Bouix, and M. Niethammer, “Gradicon: Approximate diffeomorphisms via gradient inverse consistency,” in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognitio...

  75. [83]

    Icon: Learning regular maps through inverse consistency,

    H. Greer, R. Kwitt, F.-X. Vialard, and M. Niethammer, “Icon: Learning regular maps through inverse consistency,” inProceed- ings of the IEEE/CVF International Conference on Computer Vision, 2021, pp. 3396–3405

  76. [84]

    U-net: Convolutional networks for biomedical image segmentation,

    O. Ronneberger, P. Fischer, and T. Brox, “U-net: Convolutional networks for biomedical image segmentation,” inMedical image computing and computer-assisted intervention–MICCAI 2015: 18th international conference, Munich, Germany, October 5-9, 2015, proceedings, part III 18 . S...

  77. [85]

    High- performance large-scale image recognition without normaliza- tion,

    A. Brock, S. De, S. L. Smith, and K. Simonyan, “High- performance large-scale image recognition without normaliza- tion,” in International conference on machine learning . PMLR, 2021, pp. 1059–1071

  78. [86]

    Sharpness-aware minimiza- tion improves language model generalization,

    D. Bahri, H. Mobahi, and Y . Tay, “Sharpness-aware minimiza- tion improves language model generalization,” arXiv preprint arXiv:2110.08529, 2021

  79. [87]

    Genetic epidemiology of copd (copdgene) study design,

    E. A. Regan, J. E. Hokanson, J. R. Murphy, B. Make, D. A. Lynch, T. H. Beaty, D. Curran-Everett, E. K. Silverman, and J. D. Crapo, “Genetic epidemiology of copd (copdgene) study design,” COPD: Journal of Chronic Obstructive Pulmonary Disease, vol. 7, no. 1, pp. 32–43, 2011

  80. [88]

    The osteoarthritis initia- tive,

    M. Nevitt, D. Felson, and G. Lester, “The osteoarthritis initia- tive,” Protocol for the cohort study , vol. 1, p. 2, 2006

  81. [89]

    The human connectome project: a data acquisi- tion perspective,

    D. C. Van Essen, K. Ugurbil, E. Auerbach, D. Barch, T. E. Behrens, R. Bucholz, A. Chang, L. Chen, M. Corbetta, S. W. Curtiss et al. , “The human connectome project: a data acquisi- tion perspective,” Neuroimage, vol. 62, no. 4, pp. 2222–2231, 2012

  82. [90]

    Evaluation of six registration methods for the human abdomen on clinically acquired ct,

    Z. Xu, C. P. Lee, M. P. Heinrich, M. Modat, D. Rueckert, S. Ourselin, R. G. Abramson, and B. A. Landman, “Evaluation of six registration methods for the human abdomen on clinically acquired ct,” IEEE Transactions on Biomedical Engineering , vol. 63, no. 8, pp. 1563–1572, 2016

  83. [91]

    Deep learning techniques for automatic mri cardiac multi-structures segmentation and diagnosis: is the problem solved?

    O. Bernard et al. , “Deep learning techniques for automatic mri cardiac multi-structures segmentation and diagnosis: is the problem solved?” IEEE TMI , 2018

  84. [92]

    Comparison and evaluation of methods for liver segmentation from ct datasets,

    T. Heimann et al. , “Comparison and evaluation of methods for liver segmentation from ct datasets,” IEEE TMI , 2009

  85. [93]

    Learn2reg: comprehensive multi-task medical image registration challenge, dataset and evaluation in the era of deep learning,

    A. Hering et al., “Learn2reg: comprehensive multi-task medical image registration challenge, dataset and evaluation in the era of deep learning,” IEEE TMI , 2022

  86. [94]

    Transmatch: a transformer- based multilevel dual-stream feature matching network for unsupervised deformable image registration,

    Z. Chen, Y . Zheng, and J. C. Gee, “Transmatch: a transformer- based multilevel dual-stream feature matching network for unsupervised deformable image registration,” IEEE transactions on medical imaging , vol. 43, no. 1, pp. 15–27, 2023

Pith tools

Reviewed August 15, 2026 · model on record in the stance chip above.