Skip to main navigation Skip to search Skip to main content

KDVGG-Lite: A Distilled Approach for Enhancing the Accuracy of Image Classification

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

In recent years, there has been a growing focus on the development of compact and efficient network techniques in the computer vision research field. Towards this goal, this study presents KDVGG-Lite,an innovative image classification model that is inspired from the state of-the-art VGG16 architecture. KDVGG-Lite utilizes knowledge distillation along with pruning technique from the intricate VGG16 model, attaining exceptional results on CIFAR-10 and Fashion-MNIST datasets. Knowledge distillation fusion guarantees efficient compression of essential information, while pruning enhances the model’s design for resource constrained environments. The results demonstrate the exceptional accuracy(an increase of 11.89% and 11.28%), precision (11.51% and 10.06%),recall (11.89% and 11.28%) and F1 score (12.38% and 11.40%) achieved by KDVGG-Lite on CIFAR-10 and Fashion-MNIST respectively, despite having almost 16 times fewer parameters, surpassing its VGG16 counterpart.

Original languageEnglish
Title of host publicationIntelligent Information and Database Systems - 16th Asian Conference, ACIIDS 2024, Proceedings
EditorsNgoc Thanh Nguyen, Krystian Wojtkiewicz, Richard Chbeir, Yannis Manolopoulos, Hamido Fujita, Tzung-Pei Hong, Le Minh Nguyen
PublisherSpringer Science and Business Media Deutschland GmbH
Pages15-27
Number of pages13
ISBN (Print)9789819749843
DOIs
Publication statusPublished - 2024
Event16th Asian Conference on Intelligent Information and Database Systems, ACIIDS 2024 - Ras Al Khaimah, United Arab Emirates
Duration: 15 Apr 202418 Apr 2024

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume14796 LNAI
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference16th Asian Conference on Intelligent Information and Database Systems, ACIIDS 2024
Country/TerritoryUnited Arab Emirates
CityRas Al Khaimah
Period15/04/2418/04/24

UN SDGs

This output contributes to the following UN Sustainable Development Goals (SDGs)

  1. SDG 3 - Good Health and Well-being
    SDG 3 Good Health and Well-being
  2. SDG 7 - Affordable and Clean Energy
    SDG 7 Affordable and Clean Energy
  3. SDG 9 - Industry, Innovation, and Infrastructure
    SDG 9 Industry, Innovation, and Infrastructure
  4. SDG 11 - Sustainable Cities and Communities
    SDG 11 Sustainable Cities and Communities
  5. SDG 12 - Responsible Consumption and Production
    SDG 12 Responsible Consumption and Production
  6. SDG 13 - Climate Action
    SDG 13 Climate Action
  7. SDG 17 - Partnerships for the Goals
    SDG 17 Partnerships for the Goals

Keywords

  • CIFAR-10
  • CNNs
  • Computer Vision
  • Fashion-MNIST
  • Image Classification
  • Knowledge Distillation
  • Lightweight CNN
  • VGG

Fingerprint

Dive into the research topics of 'KDVGG-Lite: A Distilled Approach for Enhancing the Accuracy of Image Classification'. Together they form a unique fingerprint.

Cite this