Design and implementation of image classification method based on deep metric learning

Image classification is a critical area of research in computer vision and deep learning, with applications in fields such as image segmentation and medical tumor detection. Despite significant advancements, challenges remain in improving classification accuracy and generalization. This paper introduces a novel deep metric learning-based image classification method to address these challenges. The model combines a CNN-based feature extraction module with a metric learning module that maps input vectors into a space where inter-class distances are maximized and intra-class distances are minimized. We propose a novel metric loss function that integrates contrast loss and triplet loss to improve accuracy and generalization. An adaptive sampling strategy dynamically selects contrast pairs based on sample difficulty, further boosting performance. Experiments with ResNet50 and GoogLeNet demonstrate the effectiveness of the proposed loss functions and sampling strategies. Additionally, a weight_decay training strategy mitigates overfitting and stabilizes training. Multiple evaluation metrics are used to assess model performance, showing that the proposed method achieves high accuracy and robust generalization.

Authors

Institutions

Publication Details

Published
2025-04-28
DOI
https://doi.org/10.1117/12.3067627
Primary Topic
Advanced Computing and Algorithms
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Design and implementation of image classification method based on deep metric learning

Rong Fei, Pengtao Zhang
Advanced Computing and Algorithms
article

Design and implementation of image classification method based on deep metric learning

Rong Fei, Pengtao Zhang
article en

Abstract

Image classification is a critical area of research in computer vision and deep learning, with applications in fields such as image segmentation and medical tumor detection. Despite significant advancements, challenges remain in improving classification accuracy and generalization. This paper introduces a novel deep metric learning-based image classification method to address these challenges. The model combines a CNN-based feature extraction module with a metric learning module that maps input vectors into a space where inter-class distances are maximized and intra-class distances are minimized. We propose a novel metric loss function that integrates contrast loss and triplet loss to improve accuracy and generalization. An adaptive sampling strategy dynamically selects contrast pairs based on sample difficulty, further boosting performance. Experiments with ResNet50 and GoogLeNet demonstrate the effectiveness of the proposed loss functions and sampling strategies. Additionally, a weight_decay training strategy mitigates overfitting and stabilizes training. Multiple evaluation metrics are used to assess model performance, showing that the proposed method achieves high accuracy and robust generalization.

Xi'an University of Technology (CN)
Openalex Percentile: Top 88%
Advanced Computing and Algorithms
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.