IHF-Harmony: Multi-Modality Magnetic Resonance Images Harmonization using Invertible Hierarchy Flow Model
Organizations: Department of Health Technology and Informatics, The Hong Kong Polytechnic University, Hong Kong · School of Integrated Circuits and Electronics, Beijing Institute of Technology, China · Mental Health Research Center, The Hong Kong Polytechnic University, Hong Kong · Department of Biomedical Engineering, Johns Hopkins University, USA
Abstract
Retrospective MRI harmonization is limited by poor scalability across modalities and reliance on traveling subject datasets. To address these challenges, we introduce IHF-Harmony, a unified invertible hierarchy flow framework for multi-modality harmonization using unpaired data. By decomposing the translation process into reversible feature transformations, IHF-Harmony guarantees bijective mapping and lossless reconstruction to prevent anatomical distortion. Specifically, an invertible hierarchy flow (IHF) performs hierarchical subtractive coupling to progressively remove artefact-related features, while an artefact-aware normalization (AAN) employs anatomy-fixed feature modulation to accurately transfer target characteristics. Combined with anatomy and artefact consistency loss objectives, IHF-Harmony achieves high-fidelity harmonization that retains source anatomy. Experiments across multiple MRI modalities demonstrate that IHF-Harmony outperforms existing methods in both anatomical fidelity and downstream task performance, facilitating robust harmonization for large-scale multi-site imaging studies. Code is available at https://github.com/Idea89560041/IHF-Harmony.
Figures & tables
| RMSE | MS-SSIM | PSNR | LPIPS | |
| w/o & | 0.4196 0.0012 | 0.5844 0.0043 | 7.54 0.026 | 0.4761 0.0033 |
| w/o | 0.1157 0.0012 | 0.8844 0.0012 | 18.75 0.09 | 0.1416 0.0014 |
| w/o | 0.0935 0.0007 | 0.9079 0.0011 | 20.58 0.06 | 0.0570 0.0009 |
| Full model | 0.0612 0.0004 | 0.9690 0.0004 | 24.59 0.05 | 0.0406 0.0009 |