AMFF-NET: Adaptive Multi-Layer Feature Fusion Network for Medical Image Segmentation

Ruirui Ma,Shaochen Jiang,Liejun Wang
DOI: https://doi.org/10.1145/3690407.3690595
2024-01-01
Abstract:In recent years, hybrid modeling approaches that integrate U-shaped networks with Transformer architectures have demonstrated significant potential in achieving precise and automatic segmentation of medical images. Nonetheless, these models exhibit a deficiency in extracting meaningful pixel features across both spatial and channel dimensions. To address these challenges and improve the model's capability to merge local attributes with overall context, while also accurately identifying spatial and channel dependencies, we propose the AMFF-NET, a new adaptive multi-layer feature fusion network. Specifically, AMFF-NET enhances the representation of foreground information in medical images by integrating a Partial Convolution and Attention Hybrid module (PCAH) with the Transformer framework. Moreover, the incorporation of an Upsampled Partial Convolution Block (UPCB) and a Parallel Channel and Spatial Attention module (PCSA) within the decoder facilitates further refinement of foreground details, while minimizing background distractions. This method allows the model to more efficiently grasp dependencies across channel and spatial dimensions. Extensive experimentations on publicly available datasets validate the superior performance of AMFF-NET.
What problem does this paper attempt to address?