Transfer Learning for Gender and Age Prediction

Cao Hong Nga; Khai Thinh Nguyen; Nghi C. Tran; Jia Ching Wang

doi:10.1109/ICCE-Taiwan49838.2020.9258347

Transfer Learning for Gender and Age Prediction

Cao Hong Nga, Khai Thinh Nguyen, Nghi C. Tran, Jia Ching Wang

Department of Computer Science and Information Engineering

Research output: Chapter in Book/Report/Conference proceeding › Conference contribution › peer-review

5 Scopus citations

Abstract

In this work, we propose a transfer learning pipeline for gender and age prediction using images from IMDB-WIKI dataset. Firstly, we freeze all layers in pre-trained ImageNet models. Then, the models are trained for four stages with scheduled learning rates and the blocks of layers are unlocked consecutively in accordance to the schedule. We apply multi-output neural network paradigm to predict age and gender simultaneously and the final loss function is based on the combination of age and gender losses. In our approach, the model has better performance than that of the non-pre-trained model because the later stages of our models reuse features extracted from the pre-trained early stages.

Original language	English
Title of host publication	2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020
Publisher	Institute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)	9781728173993
DOIs	https://doi.org/10.1109/ICCE-Taiwan49838.2020.9258347
State	Published - 28 Sep 2020
Event	7th IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020 - Taoyuan, Taiwan Duration: 28 Sep 2020 → 30 Sep 2020

Publication series

Name	2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020

Conference

Conference	7th IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020
Country/Territory	Taiwan
City	Taoyuan
Period	28/09/20 → 30/09/20

Keywords

age and gender prediction
convolutional neural network
neural networks
transfer learning

Access to Document

10.1109/ICCE-Taiwan49838.2020.9258347

Cite this

Nga, C. H., Nguyen, K. T., Tran, N. C., & Wang, J. C. (2020). Transfer Learning for Gender and Age Prediction. In 2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020 Article 9258347 (2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020). Institute of Electrical and Electronics Engineers Inc.. https://doi.org/10.1109/ICCE-Taiwan49838.2020.9258347

@inproceedings{c5e16a9395a8476cbcffb55039805952,

title = "Transfer Learning for Gender and Age Prediction",

abstract = "In this work, we propose a transfer learning pipeline for gender and age prediction using images from IMDB-WIKI dataset. Firstly, we freeze all layers in pre-trained ImageNet models. Then, the models are trained for four stages with scheduled learning rates and the blocks of layers are unlocked consecutively in accordance to the schedule. We apply multi-output neural network paradigm to predict age and gender simultaneously and the final loss function is based on the combination of age and gender losses. In our approach, the model has better performance than that of the non-pre-trained model because the later stages of our models reuse features extracted from the pre-trained early stages.",

keywords = "age and gender prediction, convolutional neural network, neural networks, transfer learning",

author = "Nga, {Cao Hong} and Nguyen, {Khai Thinh} and Tran, {Nghi C.} and Wang, {Jia Ching}",

note = "Publisher Copyright: {\textcopyright} 2020 IEEE.; 7th IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020 ; Conference date: 28-09-2020 Through 30-09-2020",

year = "2020",

month = sep,

day = "28",

doi = "10.1109/ICCE-Taiwan49838.2020.9258347",

language = "???core.languages.en_GB???",

series = "2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020",

publisher = "Institute of Electrical and Electronics Engineers Inc.",

booktitle = "2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020",

}

Nga, CH, Nguyen, KT, Tran, NC & Wang, JC 2020, Transfer Learning for Gender and Age Prediction. in 2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020., 9258347, 2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020, Institute of Electrical and Electronics Engineers Inc., 7th IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020, Taoyuan, Taiwan, 28/09/20. https://doi.org/10.1109/ICCE-Taiwan49838.2020.9258347

Transfer Learning for Gender and Age Prediction. / Nga, Cao Hong; Nguyen, Khai Thinh; Tran, Nghi C. et al.
2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020. Institute of Electrical and Electronics Engineers Inc., 2020. 9258347 (2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020).

Research output: Chapter in Book/Report/Conference proceeding › Conference contribution › peer-review

TY - GEN

T1 - Transfer Learning for Gender and Age Prediction

AU - Nga, Cao Hong

AU - Nguyen, Khai Thinh

AU - Tran, Nghi C.

AU - Wang, Jia Ching

PY - 2020/9/28

Y1 - 2020/9/28

N2 - In this work, we propose a transfer learning pipeline for gender and age prediction using images from IMDB-WIKI dataset. Firstly, we freeze all layers in pre-trained ImageNet models. Then, the models are trained for four stages with scheduled learning rates and the blocks of layers are unlocked consecutively in accordance to the schedule. We apply multi-output neural network paradigm to predict age and gender simultaneously and the final loss function is based on the combination of age and gender losses. In our approach, the model has better performance than that of the non-pre-trained model because the later stages of our models reuse features extracted from the pre-trained early stages.

AB - In this work, we propose a transfer learning pipeline for gender and age prediction using images from IMDB-WIKI dataset. Firstly, we freeze all layers in pre-trained ImageNet models. Then, the models are trained for four stages with scheduled learning rates and the blocks of layers are unlocked consecutively in accordance to the schedule. We apply multi-output neural network paradigm to predict age and gender simultaneously and the final loss function is based on the combination of age and gender losses. In our approach, the model has better performance than that of the non-pre-trained model because the later stages of our models reuse features extracted from the pre-trained early stages.

KW - age and gender prediction

KW - convolutional neural network

KW - neural networks

KW - transfer learning

UR - http://www.scopus.com/inward/record.url?scp=85098459944&partnerID=8YFLogxK

U2 - 10.1109/ICCE-Taiwan49838.2020.9258347

DO - 10.1109/ICCE-Taiwan49838.2020.9258347

M3 - 會議論文篇章

AN - SCOPUS:85098459944

T3 - 2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020

BT - 2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020

PB - Institute of Electrical and Electronics Engineers Inc.

T2 - 7th IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020

Y2 - 28 September 2020 through 30 September 2020

ER -

Nga CH, Nguyen KT, Tran NC, Wang JC. Transfer Learning for Gender and Age Prediction. In 2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020. Institute of Electrical and Electronics Engineers Inc. 2020. 9258347. (2020 IEEE International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2020). doi: 10.1109/ICCE-Taiwan49838.2020.9258347

Transfer Learning for Gender and Age Prediction

Abstract

Publication series

Conference

Keywords

Access to Document

Fingerprint

Cite this