arrow
返回

Active learning using adaptable task-based prioritisation

delete2024-07-01
delete0
delete
OA
AI
S
Shaheer U. Saeed *
J
João Ramalhinho
M
Mark A. Pinnock
Z
Ziyi Shen
Y
Yunguan Fu
M
Montana-Brown, Nina
E
Ester Bonmati
D
Dean C. Barratt
S
Stephen P. Pereira
B
Brian R Davidson
M
Matthew J. Clarkson
Y
Yipeng Hu
DOI:10.1016/j.media.2024.103181delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Supervised machine learning-based medical image computing applications necessitate expert label curation, while unlabelled image data might be relatively abundant. Active learning methods aim to prioritise a subset of available image data for expert annotation, for label-efficient model training. We develop a controller neural network that measures priority of images in a sequence of batches, as in batch-mode active learning, for multi- class segmentation tasks. The controller is optimised by rewarding positive task-specific performance gain, within a Markov decision process (MDP) environment that also optimises the task predictor. In this work, the task predictor is a segmentation network. A meta-reinforcement learning algorithm is proposed with multiple MDPs, such that the pre-trained controller can be adapted to a new MDP that contains data from different institutes and/or requires segmentation of different organs or structures within the abdomen. We present experimental results using multiple CT datasets from more than one thousand patients, with segmentation tasks of nine different abdominal organs, to demonstrate the efficacy of the learnt prioritisation controller function and its cross-institute and cross-organ adaptability. We show that the proposed adaptable prioritisation metric yields converging segmentation accuracy for a new kidney segmentation task, unseen in training, using between approximately 40% to 60% of labels otherwise required with other heuristic or random prioritisation metrics. For clinical datasets of limited size, the proposed adaptable prioritisation offers a performance improvement of 22.6% and 10.2% in Dice score, for tasks of kidney and liver vessel segmentation, respectively, compared to random prioritisation and alternative active sampling strategies.
Keyword:
Active Learning
Medical Image Quality
Segmentation

期刊

Medical Image Analysis 封面图
Medical Image Analysis
IF:
11.8
论文数:
3.9K
被引数:
2.4W

机构

U
uk research & innovation (ukri)
学者数:
2.7W
论文数: 2.3W
被引数: 32
引用论文

引用论文

A phase I study evaluating the pharmacokinetics (PK) and pharmacodynamics (PD) of the oral pan-phosphoinositide-3 kinase (PI3K) inhibitor GDC-0941
err2009-05-20
err0
PREAI
errD. Sarker; R. Kristeleit; K. E. Mazina; J. A. Ware; Y. Yan; M. Dresser; M. K. Derynck; J. De-Bono
err分享
err收藏
Automatic Multi-Organ Segmentation on Abdominal CT With Dense V-Networks基于密集V网络的腹部CT多器官自动分割
err2018-08-01
err476
errOAAI
errGibson, Eli; Giganti, Francesco; Hu, Yipeng; Bonmati, Ester; Bandula, Steve; Gurusamy, Kurinchi; Davidson, Brian; Pereira, Stephen P.; Clarkson, Matthew J.; Barratt, Dean C.
err分享
err收藏
Machine Learning for Medical Imaging1用于医学成像的机器学习1
err2017-03-01
err1.1K
errOAAI
errErickson, Bradley J.; Korfiatis, Panagiotis; Akkus, Zeynettin; Kline, Timothy L.
err分享
err收藏
KCB-Net: A 3D knee cartilage and bone segmentation network via sparse annotationKcb-net: 基于稀疏注释的3D膝关节软骨和骨分割网络
err2022-11-01
err8
errOAAI
errPeng, Yaopeng; Zheng, Hao; Liang, Peixian; Zhang, Lichun; Zaman, Fahim; Wu, Xiaodong; Sonka, Milan; Chen, Danny Z.
err分享
err收藏
学者 查看更多内容