Industry Classification of GitHub Repositories Using the North American Industry Classification System (NAICS)

· Source: cs.SE updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Software Development & Engineering, Data Science & Analytics · Depth: Expert, quick

Summary

Kevin Xu and Alexander Quispe introduce NAICS-GH, a new corpus of 6,588 GitHub repositories classified by 2-digit North American Industry Classification System (NAICS 2022) sectors. This dataset addresses the lack of standardized industry mapping for GitHub repositories, which hinders research on innovation geography and open-source production. The classification relies on a retrieve-and-verify pipeline that combines BAAI/bge-large-en embeddings, FAISS retrieval, and GPT-4.1 rubric scoring. This pipeline processed approximately 1.37 million source repositories, narrowing them to 31,178 candidates before selecting 6,588 high-confidence labels. The released labels demonstrate 96.98% precision on a 2,421-repository human-validated sample. Benchmarking six pretrained encoders on NAICS-GH shows RoBERTa-large achieving 86.45% F1 and 86.35% accuracy. The dataset, code, and fine-tuned checkpoint are publicly available.

Key takeaway

For research scientists studying innovation or open-source economics, NAICS-GH offers a critical resource for analyzing industry-specific software development. You can integrate this 6,588-repository dataset into your empirical work to map technology diffusion or industrial composition. Consider adapting the retrieve-and-verify pipeline for classifying other unstructured code data, leveraging its 96.98% precision for robust results.

Key insights

NAICS-GH provides a high-precision, AI-generated industry classification for 6,588 GitHub repositories using NAICS 2022.

Principles

Method

A retrieve-and-verify pipeline uses BAAI/bge-large-en embeddings for FAISS retrieval, followed by GPT-4.1 rubric scoring to assign 2-digit NAICS 2022 labels to GitHub repositories.

In practice

Topics

Best for: AI Scientist, Machine Learning Engineer, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.SE updates on arXiv.org.