arXiv:2507.11821cs.LGcs.AI2025-07被引 2

用智能方法自动生成树、食物等特定类别的手写图像数据集。

MNIST-Gen: A Modular MNIST-Style Dataset Generation Using Hierarchical Semantics, Reinforcement Learning, and Category Theory

  • 基于语义分层与强化学习,自动分类并生成新类别图像。
  • 生成的树和食物数据集准确率达85%,效率比人工快80%。
  • 适合需要定制数据集但无资源创建的研究者使用。

神经网络常以MNIST、FashionMNIST等标准数据集为基准,但这些数据集仅涵盖数字或衣物等通用类别,对树木、食物等具体领域任务不适用。自定义数据集的构建又耗时且受法律限制。本文提出MNIST-Gen,一个自动化、模块化、可适应的框架,能根据用户指定类别生成类MNIST图像数据集。系统结合CLIP语义理解、强化学习与人类反馈,实现低干预下的智能分类。其分层结构支持复杂语义关系,支持逐项审查、智能批量处理和快速批量生成三种模式。受范畴论启发,每个数据变换阶段被建模为可组合的态射,提升清晰度与可扩展性。作为概念验证,我们生成了两个新数据集——Tree-MNIST与Food-MNIST,证明该框架在生成任务专用评估数据上的有效性,实现85%的自动分类准确率,较人工方式节省80%时间。

原文摘要 · Abstract (English)

Neural networks are often benchmarked using standard datasets such as MNIST, FashionMNIST, or other variants of MNIST, which, while accessible, are limited to generic classes such as digits or clothing items. For researchers working on domain-specific tasks, such as classifying trees, food items, or other real-world objects, these data sets are insufficient and irrelevant. Additionally, creating and publishing a custom dataset can be time consuming, legally constrained, or beyond the scope of individual projects. We present MNIST-Gen, an automated, modular, and adaptive framework for generating MNIST-style image datasets tailored to user-specified categories using hierarchical semantic categorization. The system combines CLIP-based semantic understanding with reinforcement learning and human feedback to achieve intelligent categorization with minimal manual intervention. Our hierarchical approach supports complex category structures with semantic characteristics, enabling fine-grained subcategorization and multiple processing modes: individual review for maximum control, smart batch processing for large datasets, and fast batch processing for rapid creation. Inspired by category theory, MNIST-Gen models each data transformation stage as a composable morphism, enhancing clarity, modularity, and extensibility. As proof of concept, we generate and benchmark two novel datasets-\textit{Tree-MNIST} and \textit{Food-MNIST}-demonstrating MNIST-Gen's utility for producing task-specific evaluation data while achieving 85\% automatic categorization accuracy and 80\% time savings compared to manual approaches.

数据集生成自动化分类语义建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。