DAW-FA: Domain-aware adaptive weighting with fine-grain attention for unsupervised MRI harmonization

Autor: Linda Delali Fiasam, Yunbo Rao, Collins Sey, Stacy E.B. Aggrey, Seth Larweh Kodjiku, Kwame Opuni-Boachie Obour Agyekum, Annicet Razafindratovolahy, Isaac Adjei-Mensah, Chiagoziem Chima Ukwuoma, Francis Sam
Jazyk: angličtina
Rok vydání: 2024
Předmět:
Zdroj: Journal of King Saud University: Computer and Information Sciences, Vol 36, Iss 7, Pp 102157- (2024)
Druh dokumentu: article
ISSN: 1319-1578
DOI: 10.1016/j.jksuci.2024.102157
Popis: Magnetic resonance (MR) imaging often lacks standardized acquisition protocols across various sites, leading to contrast variations that reduce image quality and hinder automated analysis. MR harmonization improves consistency by integrating data from multiple sources, ensuring reproducible analysis. Recent advances leverage image-to-image translation and disentangled representation learning to decompose anatomical and contrast representations, achieving consistent cross-site harmonization. However, these methods face two significant drawbacks: imbalanced contrast availability during training affects adaptation performance, and insufficient utilization of spatial variability in local anatomical structures limits model adaptability to different sites. To address these challenges, we propose Domain-aware Adaptive Weighting with Fine-Grain Attention (DAW-FA) for Unsupervised MRI Harmonization. DAW-FA incorporates an adaptive weighting mechanism and enhanced self-attention to mitigate MR contrast imbalance during training and account for spatial variability in local anatomical structures. This facilitates robust cross-site harmonization without requiring paired inter-site images. We evaluated DAW-FA on MR datasets with varying scanners and acquisition protocols. Experimental results show DAW-FA outperforms existing methods, with an average increase of 1.92 ± 0.56 in Peak Signal-to-Noise Ratio (PSNR) and 0.023 ± 0.011 in Structural Similarity Index Measure (SSIM). Additionally, we demonstrate DAW-FA’s impact on downstream tasks: Alzheimer’s disease classification and whole-brain segmentation, highlighting its potential clinical relevance.
Databáze: Directory of Open Access Journals