BSAFusion: A Bidirectional Stepwise Feature Alignment Network for Unaligned Medical Image Fusion

December 11, 2024 Β· Entered Twilight Β· πŸ› AAAI Conference on Artificial Intelligence

πŸ’€ TWILIGHT: Eternal Rest
Repo abandoned since publication

Repo contents: BSAFusion.png, BSAFusion_test.py, BSAFusion_train.py, README.md, checkpoint, data, dataset, fusion_result, modal_2d, utils_2d

Authors Huafeng Li, Dayong Su, Qing Cai, Yafei Zhang arXiv ID 2412.08050 Category eess.IV: Image & Video Processing Cross-listed cs.CV, cs.LG Citations 25 Venue AAAI Conference on Artificial Intelligence Repository https://github.com/slrl123/BSAFusion ⭐ 81 Last Checked 1 month ago
Abstract
If unaligned multimodal medical images can be simultaneously aligned and fused using a single-stage approach within a unified processing framework, it will not only achieve mutual promotion of dual tasks but also help reduce the complexity of the model. However, the design of this model faces the challenge of incompatible requirements for feature fusion and alignment; specifically, feature alignment requires consistency among corresponding features, whereas feature fusion requires the features to be complementary to each other. To address this challenge, this paper proposes an unaligned medical image fusion method called Bidirectional Stepwise Feature Alignment and Fusion (BSFA-F) strategy. To reduce the negative impact of modality differences on cross-modal feature matching, we incorporate the Modal Discrepancy-Free Feature Representation (MDF-FR) method into BSFA-F. MDF-FR utilizes a Modality Feature Representation Head (MFRH) to integrate the global information of the input image. By injecting the information contained in MFRH of the current image into other modality images, it effectively reduces the impact of modality differences on feature alignment while preserving the complementary information carried by different images. In terms of feature alignment, BSFA-F employs a bidirectional stepwise alignment deformation field prediction strategy based on the path independence of vector displacement between two points. This strategy solves the problem of large spans and inaccurate deformation field prediction in single-step alignment. Finally, Multi-Modal Feature Fusion block achieves the fusion of aligned features. The experimental results across multiple datasets demonstrate the effectiveness of our method. The source code is available at https://github.com/slrl123/BSAFusion.
Community shame:
Not yet rated
Community Contributions

Found the code? Know the venue? Think something is wrong? Let us know!

πŸ“œ Similar Papers

In the same crypt β€” Image & Video Processing