IMDB-WIKI é um conjunto de dados em larga escala para estimativa de idade e gênero facial, contendo 523.051 imagens de rostos coletadas do Internet Movie Database (IMDB) e da Wikipédia. Cada imagem é anotada com um rótulo de gênero e um valor de idade derivado dos metadados associados, como datas de lançamento de filmes ou informações biográficas. O conjunto de dados foi introduzido em 2015 pelos pesquisadores Rasmus Rothe, Radu Timofte e Luc Van Gool na ETH Zurique, e tornou-se um padrão de referência para avaliar modelos de visão computacional no reconhecimento de atributos demográficos.
O conjunto de dados é notável por seu tamanho e diversidade do mundo real, pois as imagens são extraídas de fotos de celebridades e retratos biográficos. No entanto, os rótulos de idade são ruidosos porque são estimados a partir da diferença entre a data de captura da imagem e a data de nascimento do sujeito, o que pode ser impreciso para fotos mais antigas ou quando a data de captura é desconhecida. Apesar disso, o IMDB-WIKI tem sido fundamental para o avanço de abordagens de aprendizado profundo para estimativa de idade, servindo frequentemente como um corpus de pré-treinamento antes do ajuste fino em conjuntos de dados mais limpos.
Construção e Anotação
O conjunto de dados IMDB-WIKI foi construído por meio da coleta automática de imagens do IMDB e da Wikipédia. Para o IMDB, as imagens foram associadas a atores e atrizes, e a idade foi calculada como a diferença entre o ano de lançamento do filme e o ano de nascimento do ator. Para a Wikipédia, as imagens foram vinculadas a páginas biográficas, e a idade foi derivada da data da última modificação da página ou da data de nascimento do sujeito. O gênero foi atribuído com base no sexo conhecido do sujeito. O conjunto final inclui 460.723 imagens do IMDB e 62.328 da Wikipédia, totalizando 523.051 imagens. Os rostos foram detectados usando um detector de faces, e cada imagem foi recortada para a região facial, embora alguns recortes contenham erros ou múltiplos rostos.
Uso em Aprendizado Profundo
O IMDB-WIKI é amplamente utilizado na comunidade de aprendizado profundo para treinar e avaliar redes neurais em tarefas de classificação de idade e gênero. Muitos modelos de última geração, incluindo aqueles baseados em redes residuais e outras arquiteturas convolucionais, foram pré-treinados no IMDB-WIKI. A grande escala do conjunto de dados ajuda os modelos a aprender características robustas para a aparência facial, mas o ruído nos rótulos frequentemente exige funções de perda robustas ou estratégias de limpeza de dados. Pesquisadores também usaram o IMDB-WIKI para estudar os efeitos de erros de estimativa de idade, mudança de domínio e imparcialidade entre grupos demográficos.
Benchmarks e Avaliação
O IMDB-WIKI é comumente usado como um benchmark para estimativa de idade, com métricas como erro absoluto médio (MAE) e pontuação cumulativa. O conjunto de dados é dividido em conjuntos de treinamento e teste, embora a divisão oficial nem sempre seja consistente entre os estudos. Muitos artigos relatam resultados no conjunto de teste do IMDB-WIKI após ajuste fino em conjuntos de dados mais limpos, como Adience ou UTKFace. Os rótulos ruidosos tornam o benchmark desafiador, e modelos que alcançam baixo MAE no IMDB-WIKI frequentemente generalizam bem para outras tarefas de estimativa de idade. A precisão da classificação de gênero no IMDB-WIKI também é relatada, tipicamente excedendo 95% para modelos modernos.
Limitações e Considerações Éticas
O conjunto de dados tem várias limitações. Os rótulos de idade são ruidosos e tendenciosos em relação a celebridades, que podem ter fotos profissionais e maquiagem diferentes da população em geral. Os rótulos de gênero são binários e não consideram identidades não binárias. Além disso, o conjunto de dados foi coletado sem consentimento explícito dos indivíduos, levantando preocupações de privacidade. Pesquisadores que usam o IMDB-WIKI devem estar cientes dessas questões e considerar as implicações éticas de implantar modelos treinados nele, especialmente em aplicações sensíveis como vigilância ou perfilamento demográfico. O conjunto de dados está disponível para fins de pesquisa, mas seu uso em produtos comerciais pode exigir escrutínio adicional.
Conjuntos de Dados Relacionados e Impacto
O IMDB-WIKI inspirou vários conjuntos de dados subsequentes, como as versões IMDB-Clean e Wiki-Clean que filtram rótulos ruidosos, e é frequentemente comparado com o conjunto de dados Adience, que tem grupos de idade mais precisos, mas menos imagens. O conjunto de dados também foi usado em estudos sobre aumento de dados e aprendizado curricular para melhorar a precisão da estimativa de idade. Seu impacto se estende além da visão computacional, pois foi usado em pesquisas de inteligência artificial sobre imparcialidade e viés na análise facial. O conjunto de dados continua sendo um recurso-chave para pesquisadores que trabalham no reconhecimento de atributos demográficos, apesar de suas limitações conhecidas.
Ver Também
- aprendizado de máquina
- visão computacional
- reconhecimento facial
Referências
- 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.