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

Wave-GMS:用于医学图像分割的轻量级多尺度生成模型

Wave-GMS: Lightweight Multi-Scale Generative Model for Medical Image Segmentation

School of Science and Engineering, Lahore University of Management Sciences, Lahore, Pakistan(科学与工程学院,拉合尔管理科学大学,巴基斯坦拉合尔)

Talha Ahmed, Nehal Ahmed Shaikh, Hassan Mohy-ud-Din

AI总结:

本文提出轻量级多尺度生成模型Wave-GMS,以约2.6M参数实现低成本GPU上的大批量医学图像分割训练,并在四个公开数据集上取得SOTA性能与优异跨域泛化。

摘要:

为了在医院和医疗机构中公平部署AI工具,我们需要兼具高性能、并能在显存有限且成本低廉的GPU上以大批量进行训练的深度分割网络。本文提出Wave-GMS,这是一种用于医学图像分割的轻量且高效的多尺度生成模型。Wave-GMS的可训练参数量显著更少,无需加载占用大量显存的预训练视觉基础模型,并支持在显存有限的GPU上使用大批量训练。我们在四个公开数据集(BUS、BUSI、Kvasir-Instrument和HAM10000)上进行了大量实验,结果表明Wave-GMS达到了最先进的分割性能,并具有更优的跨域泛化能力,而仅需约2.6M个可训练参数。代码可在https://github.com/ATPLab-LUMS/Wave-GMS获取。

英文摘要:

For equitable deployment of AI tools in hospitals and healthcare facilities, we need Deep Segmentation Networks that offer high performance and can be trained on cost-effective GPUs with limited memory and large batch sizes. In this work, we propose Wave-GMS, a lightweight and efficient multi-scale generative model for medical image segmentation. Wave-GMS has a substantially smaller number of trainable parameters, does not require loading memory-intensive pretrained vision foundation models, and supports training with large batch sizes on GPUs with limited memory. We conducted extensive experiments on four publicly available datasets (BUS, BUSI, Kvasir-Instrument, and HAM10000), demonstrating that Wave-GMS achieves state-of-the-art segmentation performance with superior cross-domain generalizability, while requiring only ~2.6M trainable parameters. Code is available at https://github.com/ATPLab-LUMS/Wave-GMS.

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

↑