English
Related papers

Related papers: Boundary loss for highly unbalanced segmentation

200 papers

The loss function is crucial to machine learning, especially in supervised learning frameworks. It is a fundamental component that controls the behavior and general efficacy of learning algorithms. However, despite their widespread use,…

Machine Learning · Computer Science 2026-02-09 Soumi Mahato , Lineesh M. C

We introduce a novel loss function for training deep learning architectures to perform classification. It consists in minimizing the smoothness of label signals on similarity graphs built at the output of the architecture. Equivalently, it…

Machine Learning · Computer Science 2019-05-02 Myriam Bontonou , Carlos Lassance , Ghouthi Boukli Hacene , Vincent Gripon , Jian Tang , Antonio Ortega

Feed-forward CNNs trained for image transformation problems rely on loss functions that measure the similarity between the generated image and a target image. Most of the common loss functions assume that these images are spatially aligned…

Computer Vision and Pattern Recognition · Computer Science 2018-07-19 Roey Mechrez , Itamar Talmi , Lihi Zelnik-Manor

Convolutional Neural Network (CNN) have been widely used in image classification. Over the years, they have also benefited from various enhancements and they are now considered as state of the art techniques for image like data. However,…

Computer Vision and Pattern Recognition · Computer Science 2022-06-06 Thomas Gonzalez , Antoine Blais , Nicolas Couëllan , Christian Ruiz

In image segmentation, preserving the topology of segmented structures like vessels, membranes, or roads is crucial. For instance, topological errors on road networks can significantly impact navigation. Recently proposed solutions are loss…

Computer Vision and Pattern Recognition · Computer Science 2025-06-16 Benedict Schacht , Imke Greving , Simone Frintrop , Berit Zeller-Plumhoff , Christian Wilms

We proposed the boundary-integral type neural networks (BINN) for the boundary value problems in computational mechanics. The boundary integral equations are employed to transfer all the unknowns to the boundary, then the unknowns are…

Machine Learning · Computer Science 2023-05-26 Jia Sun , Yinghua Liu , Yizheng Wang , Zhenhan Yao , Xiaoping Zheng

Deep-learning architectures for classification problems involve the cross-entropy loss sometimes assisted with auxiliary loss functions like center loss, contrastive loss and triplet loss. These auxiliary loss functions facilitate better…

Computer Vision and Pattern Recognition · Computer Science 2020-04-22 Hongjun Choi , Anirudh Som , Pavan Turaga

Recent works have studied implicit biases in deep learning, especially the behavior of last-layer features and classifier weights. However, they usually need to simplify the intermediate dynamics under gradient flow or gradient descent due…

Machine Learning · Computer Science 2023-12-14 Xiong Zhou , Xianming Liu , Hanzhang Wang , Deming Zhai , Junjun Jiang , Xiangyang Ji

Fully convolutional deep neural networks have been asserted to be fast and precise frameworks with great potential in image segmentation. One of the major challenges in training such networks raises when data is unbalanced, which is common…

Computer Vision and Pattern Recognition · Computer Science 2018-12-17 Seyed Raein Hashemi , Seyed Sadegh Mohseni Salehi , Deniz Erdogmus , Sanjay P. Prabhu , Simon K. Warfield , Ali Gholipour

In machine learning, a loss function measures the difference between model predictions and ground-truth (or target) values. For neural network models, visualizing how this loss changes as model parameters are varied can provide insights…

Deep learning algorithms can fare poorly when the training dataset suffers from heavy class-imbalance but the testing criterion requires good generalization on less frequent classes. We design two novel methods to improve performance in…

Machine Learning · Computer Science 2019-10-29 Kaidi Cao , Colin Wei , Adrien Gaidon , Nikos Arechiga , Tengyu Ma

Separating overlapped nuclei is a major challenge in histopathology image analysis. Recently published approaches have achieved promising overall performance on nuclei segmentation; however, their performance on separating overlapped nuclei…

Image and Video Processing · Electrical Eng. & Systems 2021-10-01 Haotian Wang , Aleksandar Vakanski , Changfa Shi , Min Xian

The inability to guarantee robustness is one of the major obstacles to the application of deep learning models in security-demanding domains. We identify that the most commonly used cross-entropy (CE) loss does not guarantee robust boundary…

Machine Learning · Computer Science 2023-02-21 Weiwei Gao , Dazhi Zhang , Yao Li , Zhichang Guo , Ovanes Petrosian

This paper investigates the combination of intensity-based distance maps with boundary loss for point-supervised semantic segmentation. By design the boundary loss imposes a stronger penalty on the false positives the farther away from the…

Computer Vision and Pattern Recognition · Computer Science 2023-11-08 Eva Breznik , Hoel Kervadec , Filip Malmberg , Joel Kullberg , Håkan Ahlström , Marleen de Bruijne , Robin Strand

We propose a novel loss function for imbalanced classification. LDAM loss, which minimizes a margin-based generalization bound, is widely utilized for class-imbalanced image classification. Although, by using LDAM loss, it is possible to…

Computer Vision and Pattern Recognition · Computer Science 2023-06-16 Sota Kato , Kazuhiro Hotta

Generating highly detailed, complex data is a long-standing and frequently considered problem in the machine learning field. However, developing detail-aware generators remains an challenging and open problem. Generative adversarial…

Machine Learning · Computer Science 2022-09-07 Lukas Prantl , Jan Bender , Tassilo Kugelstadt , Nils Thuerey

Deep convolutional neural networks (CNN) have proven to be remarkably effective in semantic segmentation tasks. Most popular loss functions were introduced targeting improved volumetric scores, such as the Dice coefficient (DSC). By design,…

Convolutional neural networks for semantic segmentation suffer from low performance at object boundaries. In medical imaging, accurate representation of tissue surfaces and volumes is important for tracking of disease biomarkers such as…

Image and Video Processing · Electrical Eng. & Systems 2019-08-13 Francesco Caliva , Claudia Iriondo , Alejandro Morales Martinez , Sharmila Majumdar , Valentina Pedoia

Today, deep convolutional neural networks (CNNs) have demonstrated state of the art performance for supervised medical image segmentation, across various imaging modalities and tasks. Despite early success, segmentation networks may still…

Computer Vision and Pattern Recognition · Computer Science 2020-11-24 Rosana El Jurdi , Caroline Petitjean , Paul Honeine , Veronika Cheplygina , Fahed Abdallah

In neural networks, the loss function represents the core of the learning process that leads the optimizer to an approximation of the optimal convergence error. Convolutional neural networks (CNN) use the loss function as a supervisory…

Computer Vision and Pattern Recognition · Computer Science 2020-09-30 Riccardo La Grassa , Ignazio Gallo , Nicola Landro