arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.09682eess.IVcs.CV

用于组织学全切片图像分类的双重注意力模型与强化学习

Dual Attention Model with Reinforcement Learning for Classification of Histology Whole-Slide Images

University of Warwick(华威大学);The Alan Turing Institute(艾伦·图灵研究所)

Manahil Raza, Ruqayya Awan, Raja Muhammad Saad Bashir, Talha Qaiser, Nasir M. Rajpoot

AI总结:

本文提出一种双重注意力模型,结合软注意力定位感兴趣区域和基于强化学习的硬注意力多分辨率瞥见分类,在仅处理不到10%最高倍率WSI的情况下,于乳腺癌HER2评分和结直肠癌MMR状态预测任务中达到优于或可比于最先进方法的性能,并将推理时间减少75%以上。

摘要:

数字全切片图像(WSIs)通常以显微分辨率采集,包含广泛的空间数据。由于内存限制,直接将这些图像输入深度学习模型在计算上不可行,而对WSIs进行下采样则可能造成信息损失。另一种做法是将WSIs切分成较小的图像块,但这可能导致重要上下文信息的丢失。本文提出一种新颖的双重注意力方法,包含两个主要组件,均受病理学家视觉检查过程的启发:第一个软注意力模型处理WSI的低倍率视图以识别相关感兴趣区域(ROIs),随后采用一种定制的采样方法从选定的ROIs中提取多样化且空间上不同的图像瓦片。第二个组件,即硬注意力分类模型,进一步从每个瓦片中提取多分辨率瞥见序列用于分类。由于硬注意力不可微分,我们使用强化学习训练该组件以预测瞥见的位置。这种方法使模型能够聚焦于关键区域,而不是处理整个瓦片,从而与病理学家的诊断方式保持一致。两个组件通过联合损失函数以端到端方式进行训练,以证明模型的有效性。所提出的模型在两个WSI级分类问题上进行了评估:乳腺癌组织学图像上的人类表皮生长因子受体2评分,以及结直肠癌组织学图像上两种错配修复生物标志物的Intact/Loss状态预测。我们表明,所提出的模型在最高放大倍率下处理不到10%的WSI,并将推断WSI级标签所需的时间减少了75%以上,同时取得了优于或可比于最先进方法的性能。

英文摘要:

Digital whole slide images (WSIs) are generally captured at microscopic resolution and encompass extensive spatial data. Directly feeding these images to deep learning models is computationally intractable due to memory constraints, while downsampling the WSIs risks incurring information loss. Alternatively, splitting the WSIs into smaller patches may result in a loss of important contextual information. In this paper, we propose a novel dual attention approach, consisting of two main components, both inspired by the visual examination process of a pathologist: The first soft attention model processes a low magnification view of the WSI to identify relevant regions of interest, followed by a custom sampling method to extract diverse and spatially distinct image tiles from the selected ROIs. The second component, the hard attention classification model further extracts a sequence of multi-resolution glimpses from each tile for classification. Since hard attention is non-differentiable, we train this component using reinforcement learning to predict the location of the glimpses. This approach allows the model to focus on essential regions instead of processing the entire tile, thereby aligning with a pathologist's way of diagnosis. The two components are trained in an end-to-end fashion using a joint loss function to demonstrate the efficacy of the model. The proposed model was evaluated on two WSI-level classification problems: Human epidermal growth factor receptor 2 scoring on breast cancer histology images and prediction of Intact/Loss status of two Mismatch Repair biomarkers from colorectal cancer histology images. We show that the proposed model achieves performance better than or comparable to the state-of-the-art methods while processing less than 10% of the WSI at the highest magnification and reducing the time required to infer the WSI-level label by more than 75%.

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

↑