IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v12y2024i17p2647-d1464306.html
   My bibliography  Save this article

Knowledge Distillation for Enhanced Age and Gender Prediction Accuracy

Author

Listed:
  • Seunghyun Kim

    (Department of AI & Informatics, Graduate School, Sangmyung University, Seoul 03016, Republic of Korea
    These authors contributed equally to this work.)

  • Yeongje Park

    (Department of AI & Informatics, Graduate School, Sangmyung University, Seoul 03016, Republic of Korea
    These authors contributed equally to this work.)

  • Eui Chul Lee

    (Department of Human-Centered Artificial Intelligence, Sangmyung University, Seoul 03016, Republic of Korea)

Abstract

In recent years, the ability to accurately predict age and gender from facial images has gained significant traction across various fields such as personalized marketing, human–computer interaction, and security surveillance. However, the high computational cost of the current models limits their practicality for real-time applications on resource-constrained devices. This study addressed this challenge by leveraging knowledge distillation to develop lightweight age and gender prediction models that maintain a high accuracy. We propose a knowledge distillation method using teacher bounds for the efficient learning of small models for age and gender. This method allows the student model to selectively receive the teacher model’s knowledge, preventing it from unconditionally learning from the teacher in challenging age/gender prediction tasks involving factors like illusions and makeup. Our experiments used MobileNetV3 and EfficientFormer as the student models and Vision Outlooker (VOLO)-D1 as the teacher model, resulting in substantial efficiency improvements. MobileNetV3-Small, one of the student models we experimented with, achieved a 94.27% reduction in parameters and a 99.17% reduction in Giga Floating Point Operations per Second (GFLOPs). Furthermore, the distilled MobileNetV3-Small model improved gender prediction accuracy from 88.11% to 90.78%. Our findings confirm that knowledge distillation can effectively enhance model performance across diverse demographic groups while ensuring efficiency for deployment on embedded devices. This research advances the development of practical, high-performance AI applications in resource-limited environments.

Suggested Citation

  • Seunghyun Kim & Yeongje Park & Eui Chul Lee, 2024. "Knowledge Distillation for Enhanced Age and Gender Prediction Accuracy," Mathematics, MDPI, vol. 12(17), pages 1-15, August.
  • Handle: RePEc:gam:jmathe:v:12:y:2024:i:17:p:2647-:d:1464306
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/12/17/2647/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/12/17/2647/
    Download Restriction: no
    ---><---

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:12:y:2024:i:17:p:2647-:d:1464306. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.