arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2407.03307eess.IVcs.CV

HoloHisto:基于4K分辨率序列标记化的端到端千兆像素全切片图像分割

HoloHisto: End-to-end Gigapixel WSI Segmentation with 4K Resolution Sequential Tokenization

Nvidia(英伟达);Vanderbilt University(范德堡大学);Vanderbilt University Medical Center(范德堡大学医学中心)

Yucheng Tang, Yufan He, Vishwesh Nath, Pengfeig Guo, Ruining Deng, Tianyuan Yao, Quan Liu, Can Cui, Mengmeng Yin, Ziyue Xu, Holger Roth, Daguang Xu, Haichun Yang, Yuankai Huo

AI总结:

HoloHisto提出首个千兆像素WSI端到端分割方法,通过4K分辨率基础图像块和序列标记化机制直接处理完整全切片图像,并在新构建的小鼠肾脏KPIs数据集上显著超越此前最先进模型。

摘要:

在数字病理学中,基于深度学习的图像分割传统方法通常采用两阶段流程:首先将高分辨率全切片图像(WSI)分割成较小的图像块(例如256x256、512x512、1024x1024),随后将其重建回原始尺度。这种方法往往难以捕捉WSI的复杂细节和广阔范围。本文提出整体组织病理学(HoloHisto)分割方法,以在最大分辨率超过80,000×70,000像素的千兆像素WSI上实现端到端分割。HoloHisto从根本上将WSI分割范式转变为端到端学习方式,其特点包括:1)采用大尺寸(4K)分辨率基础图像块,以纳入更多视觉信息并实现高效处理;2)提出一种新颖的序列标记化机制,以恰当建模上下文关系并高效建模来自4K输入的丰富信息。据我们所知,HoloHisto首次提出了面向千兆像素分辨率WSI分割的整体方法,支持完整WSI及其对应千兆像素掩码的直接输入输出。在HoloHisto平台下,我们公开了一种超越超高分辨率的随机4K采样器,其提供的像素数分别是标准2D和3D图像块的31倍和10倍,以推进计算能力。为实现高效的4K分辨率密集预测,我们利用序列标记化,采用预训练图像标记器将图像特征分组为离散标记网格。为评估性能,我们的团队构建了一个新的肾脏病理图像分割(KPIs)数据集,包含来自完整小鼠肾脏的WSI级肾小球分割。结果表明,HoloHisto-4K相比此前最先进的模型取得了显著的性能提升。

英文摘要:

In digital pathology, the traditional method for deep learning-based image segmentation typically involves a two-stage process: initially segmenting high-resolution whole slide images (WSI) into smaller patches (e.g., 256x256, 512x512, 1024x1024) and subsequently reconstructing them to their original scale. This method often struggles to capture the complex details and vast scope of WSIs. In this paper, we propose the holistic histopathology (HoloHisto) segmentation method to achieve end-to-end segmentation on gigapixel WSIs, whose maximum resolution is above 80,000$\times$70,000 pixels. HoloHisto fundamentally shifts the paradigm of WSI segmentation to an end-to-end learning fashion with 1) a large (4K) resolution base patch for elevated visual information inclusion and efficient processing, and 2) a novel sequential tokenization mechanism to properly model the contextual relationships and efficiently model the rich information from the 4K input. To our best knowledge, HoloHisto presents the first holistic approach for gigapixel resolution WSI segmentation, supporting direct I/O of complete WSI and their corresponding gigapixel masks. Under the HoloHisto platform, we unveil a random 4K sampler that transcends ultra-high resolution, delivering 31 and 10 times more pixels than standard 2D and 3D patches, respectively, for advancing computational capabilities. To facilitate efficient 4K resolution dense prediction, we leverage sequential tokenization, utilizing a pre-trained image tokenizer to group image features into a discrete token grid. To assess the performance, our team curated a new kidney pathology image segmentation (KPIs) dataset with WSI-level glomeruli segmentation from whole mouse kidneys. From the results, HoloHisto-4K delivers remarkable performance gains over previous state-of-the-art models.

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

↑