bioRxivNeurosciencePreprint
Automatic Quality Control and Error Correction in MRI linear registration via a Residual Parameter Prediction Network for T1w MRI
Zhaojin CheniD, Roqaie MoqadamiD, Amelie MetziD, Walter Adame‐GonzaleziD, The Alzheimer’s Disease Neuroimaging Initiative (ADNI), Yashar ZeighamiiD, Mahsa DadariD
Abstract
Errors in linear registration can propagate to downstream nonlinear registration and bias volumetric estimations, deformation-based morphometry (DBM) and voxel-based morphometry (VBM) analyses. Subtle linear registration errors are particularly challenging as they are difficult to detect and may not result in obvious failures in nonlinear registration but still affect downstream results. Therefore, accurate identification and correction of these errors are critical. In this study, we present the Residual Affine COefficient Optimization Network (RACOON), a framework designed to identify and correct linear registration errors in T1w MRI scans registered to the MNI-ICBM152 space. RACOON's correction module achieved a residual misalignment RMSE of 0.778 mm on synthetic dataset, comparable to the variability observed among repeated QC-passed registrations using the same pipeline. For the classification module, RACOON achieved a balanced accuracy of 76.8% and a precision of 74.4%, outperforming existing state-of-the-art methods. RACOON is open source and publicly available at https://github.com/ZhaojinChen/RACOON.
- organism: human
- assay: imaging
- research focus: software / tool
- research focus: benchmarking
- approach: deep learning
- approach: machine learning (classical)
In-depth summary
Claude Opus 5.5 · xhigh effort · 8 Nov 2026 · from the full text of version 1 (16 Sept 2026) · CC BY 4.0
A 3D ResNet18 trained on synthetically misaligned T1w scans predicts residual affine parameters, correcting synthetic misalignments to 0.778 mm residual RMSE and flagging failed registrations better than RegQCNET or DARQ.
Errors in linear registration, including subtle ones that nonlinear registration only partly compensates, bias volumetric, deformation-based and voxel-based morphometry results, and manual quality control (QC) takes about 30 hours for roughly 10K registrations. Prior automated QC, per the authors, mostly uses preprocessed images, ignores defacing, targets large errors and only flags failures; RACOON runs on unpreprocessed, possibly defaced scans and also corrects failed transforms.
Key results
ShownOn held-out datasets with synthetic misalignments, the residual-parameter network leaves less than 1 mm of residual error. Iterative inference lowers it to about the level of disagreement between two QC-passed registrations of the same scans.
Model trained on 13,000 QC-passed scans: RMSE over six brain-boundary points was 0.861 ± 0.299 mm on the five independent test datasets in a single pass and 0.778 ± 0.286 mm with iterative inference (t=-8.36, p<0.001), vs 0.786 mm between two QC-passed registration runs in ADNI (the most variable of three datasets). NIFD, with the largest parameter MAE: 0.92 ± 0.35 mm. · 3.1.2 Training sample size and final model…; 3.1 Variability of real QC-passed images; 2.6 Assessing the variability of QC-passed…; 4. Discussion
ShownRACOON-C (fine-tuned from the residual-parameter network) had the highest accuracy and precision of all methods tested; RACOON-P (SVM on predicted parameters) ranked second in accuracy, roughly on par with direct classification and ahead of DARQ and RegQCNET. Authors suggest many real failures were subtle.
Five-fold averages on the held-out 10% of Dataset C (the Table 4 caption calls it Dataset D): accuracy 0.767 for RACOON-C and 0.721 for RACOON-P, ahead of direct classification, DARQ and RegQCNET; RACOON-C precision 0.744. Abstract: balanced accuracy 76.8%. Gains over direct classification of about 4.8 points (balanced accuracy) and 6.8 points (precision). · 3.3 Classification Module; 4. Discussion; 2.5 Performance Evaluation; Abstract
ShownDefacing augmentation removed the accuracy drop seen when the network was applied to a defaced independent dataset (CamCAN).
Without defacing augmentation, RMSE rose from 1.39 ± 0.60 mm on validation to 1.69 ± 0.56 mm on defaced CamCAN (t=-11.68, p<0.001). With augmentation (nine facial masks, four manual and five from AFNI, FreeSurfer, FSL, pydeface, mridefacer), RMSEs were 1.04 ± 0.37 mm and 1.03 ± 0.31 mm, not significantly different. · 4. Discussion; 2.4.2 Augmentation
SuggestedThe predicted residual parameters fixed real ANTs registration failures. In the example shown, a corrected scan passed visual QC performed blind to correction status.
Qualitative examples only (Fig. 7). One scan had a scaling error in all three views and a rotation error clearest in axial slices; after correction its agreement with the template outline improved in every view. No count or rate of corrected real failures is given. · 3.4 Correction Module
SuggestedPrediction errors show the same axis dependence as the variability between repeated registrations: more rotation error about x, less scaling and translation error along x. This hints that part of the residual error is intrinsic to affine registration rather than a model bias.
Synthetic test sets: x-axis rotation MAE 0.305° to 0.336° vs about 0.19° to 0.23° for y and z; x-axis translation MAE 0.091 to 0.139 mm vs about 0.20 to 0.26 mm for y and z. Repeated QC-passed registrations in ADNI, UKBB and PPMI showed the same pattern. · 3.1.2 Training sample size and final model…; 3.1 Variability of real QC-passed images; 4. Discussion
Methods, data and reproducibility
Rotation, scaling and translation offsets were applied to QC-passed T1w scans registered to MNI-ICBM152-2009c; a 3D ResNet18 learned to predict these residual affine parameters from unpreprocessed volumes. Predicted parameters feed an SVM pass/fail classifier (RACOON-P) and an iterative correction of the original transform; a second classifier (RACOON-C) is fine-tuned from the network's backbone on the 3D volume.
- T1w MRI from 13 public datasets (incl. ADNI, UK Biobank, PPMI, CamCAN); Siemens, GE, Philips; 1.5 T and 3.0 T; healthy aging, frontotemporal dementia, Parkinson's and Alzheimer's disease · human (clinical cohorts such as ADNI, UK Biobank) · n = 13 datasets
- Dataset B: QC-passed scans with synthetic affine misalignment for the residual-parameter network; five datasets held out as independent tests, eight form the training pool · human · n = 13,000 training scans (final model)
- Dataset A: subset for the template-input ablation and the scaling/translation loss-weight grid search · human
- Dataset C: equal numbers of real QC-passed and QC-failed scans with manual labels; 10% held out for testing, 90% for five-fold cross-validation · human
- Repeated linear registrations (two runs, both QC-passed) used to estimate acceptable affine variability · human · n = 3 datasets (ADNI, UKBB, PPMI)
- CamCAN, a defaced dataset, used to test the effect of defacing augmentation · human
- Methods & tools
- 3D ResNet18, SVM (RBF kernel), ANTs, PELICAN, MNI-ICBM152-2009c template, PyTorch, TorchIO, SimpleITK, AFNI, FreeSurfer, FSL, pydeface, mridefacer (defacing masks), RegQCNET, DARQ
- Software & environment
- PyTorch with automatic mixed precision (torch.cuda.amp.autocast) and TorchIO; no package versions, requirement files or containers stated.
- Assay conditions & compute
- Computing resources from the Digital Research Alliance of Canada; hardware, run time and memory not stated.
- Reproducibility gaps
- Trained models are released with the code. Dataset descriptions and the compositions and sizes of subsets A, B and C are deferred to Supplementary material. No data URLs or accessions are given in the text, and the release of the manual QC labels is not mentioned.
Limitations (6)
- Real QC-failed registrations came mainly from ANTs, so the range of real failure types may be narrow. The authors argue that the synthetic parameter sampling is tool-agnostic, but suggest adding failures from other registration methods. · stated by the authors
- Shearing was not among the training transformations, so registration failures caused by shearing are not corrected (Fig. S8). · stated by the authors
- Only linear registration is checked; QC of nonlinear registration, skull stripping and tissue segmentation is left for future work. · stated by the authors
- Developed and validated for MNI-ICBM152-2009c; templates whose stereotaxic alignment differs substantially would need retraining. · stated by the authors
- Correction of real failures is shown only through example scans (Fig. 7), and no proportion of real QC failures that pass after correction is reported. The headline correction accuracy therefore rests on synthetic misalignments. · noted in this summary
- The classification comparison uses one held-out split (10% of Dataset C) labelled with the authors' own QC criteria, and the test-set size is not given in the text. DARQ and RegQCNET were run as released models, with RegQCNET's threshold set by ROC analysis. · noted in this summary
Read this if you register large T1w cohorts linearly to MNI space and want automated QC that can also repair failed affine transforms instead of discarding the scans.
- Citations
- 0
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- MACHINE LEARNING FRAMEWORK FOR FULLY AUTOMATIC QUALITY CHECKING OF RIGID AND AFFINE REGISTRATIONS IN BIG DATA BRAIN MRI
Sudhakar Tummala, Niels K. Focke·IEEE International Symposium on Biomedical Imaging (ISBI)Peer-reviewed·Apr 13, 2021
- Deep learning of quality control for stereotaxic registration of human brain MRI
Vladimir Fonov, Mahsa Dadar, D. Louis Collins·bioRxiv·Apr 18, 2018
- DARQ: Deep learning of quality control for stereotaxic registration of human brain MRI
Vladimir Fonov, Mahsa Dadar, … D. Louis Collins·bioRxiv·Aug 16, 2021
- MMORF - FSL's MultiMOdal Registration Framework
Frederik Lange, Christoph Arthofer, … Jesper Andersson·Imaging NeurosciencePeer-reviewed·Jan 1, 2024
- A New Approach to Symmetric Registration of Longitudinal Structural MRI of the Human Brain
Babak Assai Ardekani, for the Alzheimer’s Disease Neuroimaging Initiative·Journal of Neuroscience MethodsPeer-reviewed·May 1, 2022
- DACO: Distortion/artefact correction for diffusion MRI data in an integrated framework
Yung‐Chin Hsu, Wen‐Yih Isaac Tseng·bioRxiv·Jul 6, 2021