IMDB-WIKI es un conjunto de datos a gran escala para la estimación de edad y género facial, que contiene 523,051 imágenes de rostros recopiladas de Internet Movie Database (IMDB) y Wikipedia. Cada imagen está anotada con una etiqueta de género y un valor de edad derivado de los metadatos asociados, como las fechas de estreno de películas o la información biográfica. El conjunto de datos fue presentado en 2015 por los investigadores Rasmus Rothe, Radu Timofte y Luc Van Gool en la ETH Zúrich, y se ha convertido en un punto de referencia estándar para evaluar modelos de visión por computadora en el reconocimiento de atributos demográficos.
El conjunto de datos destaca por su tamaño y su diversidad del mundo real, ya que las imágenes provienen de fotos de celebridades y retratos biográficos. Sin embargo, las etiquetas de edad son ruidosas porque se estiman a partir de la diferencia entre la fecha de captura de la imagen y la fecha de nacimiento del sujeto, lo que puede ser inexacto para fotos antiguas o cuando se desconoce la fecha de captura. A pesar de esto, IMDB-WIKI ha sido fundamental para avanzar en los enfoques de aprendizaje profundo para la estimación de edad, sirviendo a menudo como un corpus de preentrenamiento antes de ajustar modelos en conjuntos de datos más limpios.
Construcción y Anotación
El conjunto de datos IMDB-WIKI se construyó mediante el rastreo automático de imágenes de IMDB y Wikipedia. Para IMDB, las imágenes se asociaron con actores y actrices, y la edad se calculó como la diferencia entre el año de estreno de la película y el año de nacimiento del actor. Para Wikipedia, las imágenes se vincularon a páginas biográficas, y la edad se derivó de la fecha de última modificación de la página o de la fecha de nacimiento del sujeto. El género se asignó según el sexo conocido del sujeto. El conjunto de datos final incluye 460,723 imágenes de IMDB y 62,328 de Wikipedia, totalizando 523,051 imágenes. Los rostros se detectaron mediante un detector de rostros, y cada imagen se recortó a la región facial, aunque algunos recortes contienen errores o múltiples rostros.
Uso en Aprendizaje Profundo
IMDB-WIKI se utiliza ampliamente en la comunidad de aprendizaje profundo para entrenar y evaluar redes neuronales en tareas de clasificación de edad y género. Muchos modelos de última generación, incluidos los basados en redes residuales y otras arquitecturas convolucionales, se han preentrenado en IMDB-WIKI. La gran escala del conjunto de datos ayuda a los modelos a aprender características robustas para la apariencia facial, pero su ruido en las etiquetas a menudo requiere funciones de pérdida robustas o estrategias de limpieza de datos. Los investigadores también han utilizado IMDB-WIKI para estudiar los efectos de los errores de estimación de edad, el cambio de dominio y la equidad entre grupos demográficos.
Puntos de Referencia y Evaluación
IMDB-WIKI se utiliza comúnmente como punto de referencia para la estimación de edad, con métricas como el error absoluto medio (MAE) y la puntuación acumulativa. El conjunto de datos se divide en conjuntos de entrenamiento y prueba, aunque la división oficial no siempre es consistente entre los estudios. Muchos artículos informan resultados en el conjunto de prueba de IMDB-WIKI después de ajustar modelos en conjuntos de datos más limpios como Adience o UTKFace. Las etiquetas ruidosas lo convierten en un punto de referencia desafiante, y los modelos que logran un MAE bajo en IMDB-WIKI a menudo generalizan bien a otras tareas de estimación de edad. También se informa la precisión de clasificación de género en IMDB-WIKI, que típicamente supera el 95% para los modelos modernos.
Limitaciones y Consideraciones Éticas
El conjunto de datos tiene varias limitaciones. Las etiquetas de edad son ruidosas y están sesgadas hacia celebridades, quienes pueden tener fotos profesionales y maquillaje que difieren de la población general. Las etiquetas de género son binarias y no tienen en cuenta las identidades no binarias. Además, el conjunto de datos se recopiló sin consentimiento explícito de los individuos, lo que plantea preocupaciones de privacidad. Los investigadores que utilizan IMDB-WIKI deben ser conscientes de estos problemas y considerar las implicaciones éticas de implementar modelos entrenados en él, especialmente en aplicaciones sensibles como la vigilancia o la elaboración de perfiles demográficos. El conjunto de datos está disponible para fines de investigación, pero su uso en productos comerciales puede requerir un escrutinio adicional.
Conjuntos de Datos Relacionados e Impacto
IMDB-WIKI ha inspirado varios conjuntos de datos posteriores, como las versiones IMDB-Clean y Wiki-Clean que filtran etiquetas ruidosas, y a menudo se compara con el conjunto de datos Adience, que tiene grupos de edad más precisos pero menos imágenes. El conjunto de datos también se ha utilizado en estudios sobre aumento de datos y aprendizaje curricular para mejorar la precisión de la estimación de edad. Su impacto se extiende más allá de la visión por computadora, ya que se ha utilizado en investigaciones de inteligencia artificial sobre equidad y sesgo en el análisis facial. El conjunto de datos sigue siendo un recurso clave para los investigadores que trabajan en el reconocimiento de atributos demográficos, a pesar de sus limitaciones conocidas.
Véase También
- aprendizaje automático
- visión por computadora
- reconocimiento facial
Referencias
- Rothe, R., Timofte, R., & Van Gool, L. (2015). DEX: Deep EXpectation of apparent age from a single image. In Proceedings of the IEEE International Conference on Computer Vision Workshops.
- Rothe, R., Timofte, R., & Van Gool, L. (2016). Deep expectation of real and apparent age from a single image without facial landmarks. International Journal of Computer Vision.