BAF-Net: Bidirectional Attention-aware Fluid Pyramid Feature Integrated Multi-modal Fusion Network for Prognosis

  • Huiqin Wu
  • , Wenbing Lv
  • , Dongyang Du
  • , Hui Xu
  • , Guoyu Lin
  • , Zidong Zhou
  • , Jianhua Ma
  • , Lijun Lu

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

In this work, we proposed a bidirectional attention-aware fluid pyramid features integrated fusion network (BAF-Net) with cross-modal interactions for multi-modality medical images prognosis. The network is composed of two identical branches to preserve the unimodal feature and one paralleled bidirectional attention-aware distillation stream to assimilate cross-modal complements progressively and learn supplementary refined features in both the bottom-up and the top-down processes. The fluid pyramid connections were adopted to integrate the hierarchical features at different levels of deep neural network, and channel-wise attention modules were exploited to mitigate the cross-modal cross-level incompatibility. Furthermore, the depth-wise separable convolution was introduced to fuse the cross-modal cross-level features to alleviate the increase of parameters to a great extent. Better performance was obtained by the proposed BAF-Net compared to unimodal network for cancer prognosis in a public TCIA dataset (head & neck cancer (HNC) dataset composed of 800 patients from nine centers).

Original languageEnglish
Title of host publication2022 IEEE NSS/MIC RTSD - IEEE Nuclear Science Symposium, Medical Imaging Conference and Room Temperature Semiconductor Detector Conference
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781665488723
DOIs
StatePublished - 2022
Externally publishedYes
Event2022 IEEE Nuclear Science Symposium, Medical Imaging Conference, and Room Temperature Semiconductor Detector Conference, IEEE NSS MIC RTSD 2022 - Milano, Italy
Duration: 5 Nov 202212 Nov 2022

Publication series

Name2022 IEEE NSS/MIC RTSD - IEEE Nuclear Science Symposium, Medical Imaging Conference and Room Temperature Semiconductor Detector Conference

Conference

Conference2022 IEEE Nuclear Science Symposium, Medical Imaging Conference, and Room Temperature Semiconductor Detector Conference, IEEE NSS MIC RTSD 2022
Country/TerritoryItaly
CityMilano
Period5/11/2212/11/22

Keywords

  • Bidirectional attention-aware
  • deep learning
  • diagnosis and prognosis
  • fluid pyramid feature integration
  • multi-modalities fusion

Fingerprint

Dive into the research topics of 'BAF-Net: Bidirectional Attention-aware Fluid Pyramid Feature Integrated Multi-modal Fusion Network for Prognosis'. Together they form a unique fingerprint.

Cite this