arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2412.00351eess.IVcs.CV

用于医学图像分析的多任务学习中的多尺度特征增强

Multi-scale Feature Enhancement in Multi-task Learning for Medical Image Analysis

Phuoc-Nguyen Bui, Duc-Tai Le, Junghyun Bum, Hyunseung Choo

AI总结:

本文提出基于 UNet 的医学图像多任务模型,利用 ResFormer、多尺度特征融合与 DFE 模块同时提升分类和分割性能。

摘要:

医学影像中的传统深度学习方法通常仅专注于分割或分类,限制了其利用共享信息的能力。多任务学习(MTL)通过共享表示将这两项任务结合起来解决这一问题,但往往难以平衡用于分割的局部空间特征与用于分类的全局语义特征,从而导致性能欠佳。本文提出一种简单而有效的基于 UNet 的 MTL 模型,其中编码器提取的特征用于预测分类标签,而解码器生成分割掩码。该模型引入了先进编码器,其中包含新颖的 ResFormer 块,将卷积特征提取得到的局部上下文与 Transformer 建模的长程依赖相结合。该设计捕获更广泛的上下文关系和细粒度细节,提高分类和分割准确率。为增强分类性能,模型组合来自不同编码器层级的多尺度特征,以利用输入图像的层次化表示。对于分割,通过跳跃连接传递给解码器的特征使用新颖的扩张特征增强(DFE)模块进行细化,该模块通过三个具有不同扩张率的并行卷积分支捕获不同尺度的信息。这使解码器能够以更高准确率检测不同大小的病灶。在多个医学数据集上的实验结果证实,与最先进的单任务和多任务学习方法相比,我们的模型在分割和分类两项任务上均具有优越性能。

英文摘要:

Traditional deep learning methods in medical imaging often focus solely on segmentation or classification, limiting their ability to leverage shared information. Multi-task learning (MTL) addresses this by combining both tasks through shared representations but often struggles to balance local spatial features for segmentation and global semantic features for classification, leading to suboptimal performance. In this paper, we propose a simple yet effective UNet-based MTL model, where features extracted by the encoder are used to predict classification labels, while the decoder produces the segmentation mask. The model introduces an advanced encoder incorporating a novel ResFormer block that integrates local context from convolutional feature extraction with long-range dependencies modeled by the Transformer. This design captures broader contextual relationships and fine-grained details, improving classification and segmentation accuracy. To enhance classification performance, multi-scale features from different encoder levels are combined to leverage the hierarchical representation of the input image. For segmentation, the features passed to the decoder via skip connections are refined using a novel dilated feature enhancement (DFE) module, which captures information at different scales through three parallel convolution branches with varying dilation rates. This allows the decoder to detect lesions of varying sizes with greater accuracy. Experimental results across multiple medical datasets confirm the superior performance of our model in both segmentation and classification tasks, compared to state-of-the-art single-task and multi-task learning methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑