Deep Metric Learning by Exploring Confusing Triplet Embeddings for COVID-19 Medical Images Diagnosis

Tongtong Yuan, Lingmei Dong, Bo Liu, Jialiang Huang, Chuangbai Xiao
Proceedings of the 1st Workshop on Healthcare AI and COVID-19, ICML 2022, PMLR 184:1-10, 2022.

Abstract

Because the COVID-19 virus is highly transmissible, leading to a worldwide increment of new infections and deaths daily, the development of an automated tool to identify COVID-19 using CT images has attracted much attention. Significantly, deep metric learning can be deployed to cluster and classify the fine-grained CT images, which aims to learn a mapping from the original objects to a discriminative feature embedding space. Previous deep metric learning works have been proposed to construct various structures of loss, mine hard samples, or introduce regularization constraints, \etc. In general, traditional loss functions of deep metric learning methods are based on constraining the distance of the triplet embeddings in the feature space. Instead of focusing on the previous research directions, in this work, we pay attention to exploring confusing triplet embeddings, for the reason that confusing triplet embeddings perform a side effect on the majority of deep triplet-based metric learning methods. By considering the spatial relation of triplet embedding, and conducting theoretical analysis in the feature space, we propose an approach to recognize the confusing triplet embeddings and construct a Confusing Triplet Embedding Learning (CTEL) method by adding a hard constraint on the confusing triplet embeddings. The extensive experiments indicate that our proposed CTEL method achieves more excellent performance on two medical CT image datasets and two fine-grained standard image datasets compared with many state-of-the-art methods.

Cite this Paper


BibTeX
@InProceedings{pmlr-v184-yuan22a, title = {Deep Metric Learning by Exploring Confusing Triplet Embeddings for COVID-19 Medical Images Diagnosis}, author = {Yuan, Tongtong and Dong, Lingmei and Liu, Bo and Huang, Jialiang and Xiao, Chuangbai}, booktitle = {Proceedings of the 1st Workshop on Healthcare AI and COVID-19, ICML 2022}, pages = {1--10}, year = {2022}, editor = {Xu, Peng and Zhu, Tingting and Zhu, Pengkai and Clifton, David A. and Belgrave, Danielle and Zhang, Yuanting}, volume = {184}, series = {Proceedings of Machine Learning Research}, month = {22 Jul}, publisher = {PMLR}, pdf = {https://proceedings.mlr.press/v184/yuan22a/yuan22a.pdf}, url = {https://proceedings.mlr.press/v184/yuan22a.html}, abstract = {Because the COVID-19 virus is highly transmissible, leading to a worldwide increment of new infections and deaths daily, the development of an automated tool to identify COVID-19 using CT images has attracted much attention. Significantly, deep metric learning can be deployed to cluster and classify the fine-grained CT images, which aims to learn a mapping from the original objects to a discriminative feature embedding space. Previous deep metric learning works have been proposed to construct various structures of loss, mine hard samples, or introduce regularization constraints, \etc. In general, traditional loss functions of deep metric learning methods are based on constraining the distance of the triplet embeddings in the feature space. Instead of focusing on the previous research directions, in this work, we pay attention to exploring confusing triplet embeddings, for the reason that confusing triplet embeddings perform a side effect on the majority of deep triplet-based metric learning methods. By considering the spatial relation of triplet embedding, and conducting theoretical analysis in the feature space, we propose an approach to recognize the confusing triplet embeddings and construct a Confusing Triplet Embedding Learning (CTEL) method by adding a hard constraint on the confusing triplet embeddings. The extensive experiments indicate that our proposed CTEL method achieves more excellent performance on two medical CT image datasets and two fine-grained standard image datasets compared with many state-of-the-art methods.} }
Endnote
%0 Conference Paper %T Deep Metric Learning by Exploring Confusing Triplet Embeddings for COVID-19 Medical Images Diagnosis %A Tongtong Yuan %A Lingmei Dong %A Bo Liu %A Jialiang Huang %A Chuangbai Xiao %B Proceedings of the 1st Workshop on Healthcare AI and COVID-19, ICML 2022 %C Proceedings of Machine Learning Research %D 2022 %E Peng Xu %E Tingting Zhu %E Pengkai Zhu %E David A. Clifton %E Danielle Belgrave %E Yuanting Zhang %F pmlr-v184-yuan22a %I PMLR %P 1--10 %U https://proceedings.mlr.press/v184/yuan22a.html %V 184 %X Because the COVID-19 virus is highly transmissible, leading to a worldwide increment of new infections and deaths daily, the development of an automated tool to identify COVID-19 using CT images has attracted much attention. Significantly, deep metric learning can be deployed to cluster and classify the fine-grained CT images, which aims to learn a mapping from the original objects to a discriminative feature embedding space. Previous deep metric learning works have been proposed to construct various structures of loss, mine hard samples, or introduce regularization constraints, \etc. In general, traditional loss functions of deep metric learning methods are based on constraining the distance of the triplet embeddings in the feature space. Instead of focusing on the previous research directions, in this work, we pay attention to exploring confusing triplet embeddings, for the reason that confusing triplet embeddings perform a side effect on the majority of deep triplet-based metric learning methods. By considering the spatial relation of triplet embedding, and conducting theoretical analysis in the feature space, we propose an approach to recognize the confusing triplet embeddings and construct a Confusing Triplet Embedding Learning (CTEL) method by adding a hard constraint on the confusing triplet embeddings. The extensive experiments indicate that our proposed CTEL method achieves more excellent performance on two medical CT image datasets and two fine-grained standard image datasets compared with many state-of-the-art methods.
APA
Yuan, T., Dong, L., Liu, B., Huang, J. & Xiao, C.. (2022). Deep Metric Learning by Exploring Confusing Triplet Embeddings for COVID-19 Medical Images Diagnosis. Proceedings of the 1st Workshop on Healthcare AI and COVID-19, ICML 2022, in Proceedings of Machine Learning Research 184:1-10 Available from https://proceedings.mlr.press/v184/yuan22a.html.

Related Material