%0 Conference Paper %1 10.1007/978-3-319-67389-9_44 %A Salehi, Seyed Sadegh Mohseni %A Erdogmus, Deniz %A Gholipour, Ali %B Machine Learning in Medical Imaging %C Cham %D 2017 %E Wang, Qian %E Shi, Yinghuan %E Suk, Heung-Il %E Suzuki, Kenji %I Springer International Publishing %K loss-function order2 tversky %P 379--387 %T Tversky Loss Function for Image Segmentation Using 3D Fully Convolutional Deep Networks %X Fully convolutional deep neural networks carry out excellent potential for fast and accurate image segmentation. One of the main challenges in training these networks is data imbalance, which is particularly problematic in medical imaging applications such as lesion segmentation where the number of lesion voxels is often much lower than the number of non-lesion voxels. Training with unbalanced data can lead to predictions that are severely biased towards high precision but low recall (sensitivity), which is undesired especially in medical applications where false negatives are much less tolerable than false positives. Several methods have been proposed to deal with this problem including balanced sampling, two step training, sample re-weighting, and similarity loss functions. In this paper, we propose a generalized loss function based on the Tversky index to address the issue of data imbalance and achieve much better trade-off between precision and recall in training 3D fully convolutional deep neural networks. Experimental results in multiple sclerosis lesion segmentation on magnetic resonance images show improved \$\$F\_2\$\$score, Dice coefficient, and the area under the precision-recall curve in test data. Based on these results we suggest Tversky loss function as a generalized framework to effectively train deep neural networks. %@ 978-3-319-67389-9

@inproceedings{10.1007/978-3-319-67389-9_44, abstract = {Fully convolutional deep neural networks carry out excellent potential for fast and accurate image segmentation. One of the main challenges in training these networks is data imbalance, which is particularly problematic in medical imaging applications such as lesion segmentation where the number of lesion voxels is often much lower than the number of non-lesion voxels. Training with unbalanced data can lead to predictions that are severely biased towards high precision but low recall (sensitivity), which is undesired especially in medical applications where false negatives are much less tolerable than false positives. Several methods have been proposed to deal with this problem including balanced sampling, two step training, sample re-weighting, and similarity loss functions. In this paper, we propose a generalized loss function based on the Tversky index to address the issue of data imbalance and achieve much better trade-off between precision and recall in training 3D fully convolutional deep neural networks. Experimental results in multiple sclerosis lesion segmentation on magnetic resonance images show improved {\$}{\$}F{\_}2{\$}{\$}score, Dice coefficient, and the area under the precision-recall curve in test data. Based on these results we suggest Tversky loss function as a generalized framework to effectively train deep neural networks.}, added-at = {2020-06-02T20:26:42.000+0200}, address = {Cham}, author = {Salehi, Seyed Sadegh Mohseni and Erdogmus, Deniz and Gholipour, Ali}, biburl = {https://www.bibsonomy.org/bibtex/2da131b3b28d906dfeb5b36b73223f4f5/sohnki}, booktitle = {Machine Learning in Medical Imaging}, description = {Tversky Loss Function for Image Segmentation Using 3D Fully Convolutional Deep Networks | SpringerLink}, editor = {Wang, Qian and Shi, Yinghuan and Suk, Heung-Il and Suzuki, Kenji}, interhash = {ed483f0847e333647143d0594a6dac56}, intrahash = {da131b3b28d906dfeb5b36b73223f4f5}, isbn = {978-3-319-67389-9}, keywords = {loss-function order2 tversky}, pages = {379--387}, publisher = {Springer International Publishing}, timestamp = {2020-06-02T20:26:42.000+0200}, title = {Tversky Loss Function for Image Segmentation Using 3D Fully Convolutional Deep Networks}, year = 2017 }