Intersection-over-Union(IoUスコア)としても知られるJaccardインデックスは、医療画像のセグメンテーションにおける最も重要な評価指標の1つです。ただし、複数の目的クラスにわたって平均IoU(mIoU)スコアを直接最適化することは未解決の問題です。その代理を最適化するためにいくつかのアルゴリズムが提案されていますが、それらの一般化能力については保証が提供されていません。この論文では、厳格な下限に裏打ちされた、データ分布全体にわたるmIoUのより良い一般化のための、新しいデータ分布を意識したマージンキャリブレーション方法を提示します。このスキームにより、実際のIoUスコアの観点からより優れたセグメンテーションパフォーマンスが保証されます。 2つの医療画像セグメンテーションデータセットで提案されたマージンキャリブレーション方法の有効性を評価し、ディープセグメンテーションモデルを使用した他の学習スキームよりもIoUスコアの大幅な改善を示しています。
The Jaccard index, also known as Intersection-over-Union (IoU score), is one of the most critical evaluation metrics in medical image segmentation. However, directly optimizing the mean IoU (mIoU) score over multiple objective classes is an open problem. Although some algorithms have been proposed to optimize its surrogates, there is no guarantee provided for their generalization ability. In this paper, we present a novel data-distribution-aware margin calibration method for a better generalization of the mIoU over the whole data-distribution, underpinned by a rigid lower bound. This scheme ensures a better segmentation performance in terms of IoU scores in practice. We evaluate the effectiveness of the proposed margin calibration method on two medical image segmentation datasets, showing substantial improvements of IoU scores over other learning schemes using deep segmentation models.