Calibrated Surrogate Maximization of Dice

Nordström, MarcusKTH,Matematisk statistik,RaySearch Laboratories, Stockholm, Sweden (author)

QC 20210325
In the medical imaging community, it is increasingly popular to train machine learning models for segmentation problems with objectives based on the soft-Dice surrogate. While experimental studies have showed good performance with respect to Dice, there have also been reports of some issues related to stability. In parallel with these developments, direct optimization of evaluation metrics has also been studied in the context of binary classification. Recently, in this setting, a quasi-concave, lower-bounded and calibrated surrogate for the F1-score has been proposed. In this work, we show how to use this surrogate in the context of segmentation. We then show that it has some better theoretical properties than soft-Dice. Finally, we experimentally compare the new surrogate with soft-Dice on a 3D-segmentation problem and get results indicating that stability is improved. We conclude that the new surrogate, for theoretical and experimental reasons, can be considered a promising alternative to the soft-Dice surrogate.

Bao, H. (author)
Löfman, F. (author)
Hult, Henrik,1975-KTH,Matematisk statistik(Swepub:kth)u1q9faoy (author)
Maki, AtsutoKTH,Robotik, perception och lärande, RPL(Swepub:kth)u1elx760 (author)
Sugiyama, M. (author)
KTHMatematisk statistik (creator_code:org_t)

In:Medical Image Computing and Computer Assisted Intervention – MICCAI 2020Cham : Springer Nature, s. 269-278

By the author/editor: Nordström, Marcu ...; Bao, H.; Löfman, F.; Hult, Henrik, 19 ...; Maki, Atsuto; Sugiyama, M.

About the subject

ENGINEERING AND TECHNOLOGY: ENGINEERING AND ...; and Medical Engineer ...; and Medical Image Pr ...

Kungliga biblioteket hanterar dina personuppgifter i enlighet med EU:s dataskyddsförordning (2018), GDPR. Läs mer om hur det funkar här.
Så här hanterar KB dina uppgifter vid användning av denna tjänst.