Eigenloss: Combined PCA-Based Loss Function for Polyp Segmentation: Combined PCA-based loss function for polyp segmentation

Luisa F. Sánchez-Peralta, Artzai Picón, Juan Antonio Antequera-Barroso, Juan Francisco Ortega-Morán, Francisco M. Sánchez-Margallo, J. Blas Pagador

Research output: Contribution to journalArticlepeer-review

12 Citations (Scopus)

Abstract

Colorectal cancer is one of the leading cancer death causes worldwide, but its early diagnosis highly improves the survival rates. The success of deep learning has also benefited this clinical field. When training a deep learning model, it is optimized based on the selected loss function. In this work, we consider two networks (U-Net and LinkNet) and two backbones (VGG-16 and Densnet121). We analyzed the influence of seven loss functions and used a principal component analysis (PCA) to determine whether the PCA-based decomposition allows for the defining of the coefficients of a non-redundant primal loss function that can outperform the individual loss functions and different linear combinations. The eigenloss is defined as a linear combination of the individual losses using the elements of the eigenvector as coefficients. Empirical results show that the proposed eigenloss improves the general performance of individual loss functions and outperforms other linear combinations when Linknet is used, showing potential for its application in polyp segmentation problems.
Original languageEnglish
Article number1316
Pages (from-to)1316
Number of pages1
JournalMathematics
Volume8
Issue number8
DOIs
Publication statusPublished - Aug 2020

Keywords

  • Deep learning
  • Loss functions
  • Principal component analysis
  • Polyp segmentation

Fingerprint

Dive into the research topics of 'Eigenloss: Combined PCA-Based Loss Function for Polyp Segmentation: Combined PCA-based loss function for polyp segmentation'. Together they form a unique fingerprint.

Cite this