arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.02114cs.LGcs.AIcs.CL
收藏

Tiny-Toxic-Detector:一种用于有毒内容检测的紧凑型Transformer模型

Tiny-Toxic-Detector: A compact transformer-based model for toxic content detection

Assistantslab(助理实验室)

Michiel Kamphuis

AI总结:

本文提出仅210万参数的紧凑型Transformer模型Tiny-toxic-detector,在ToxiGen和Jigsaw数据集上以90.97%和86.98%的准确率媲美大50倍以上的模型,为资源受限环境下的内容审核提供了高效可扩展的解决方案。

摘要:

本文介绍了Tiny-toxic-detector,一种用于有毒内容检测的紧凑型Transformer模型。尽管仅有210万参数,Tiny-toxic-detector在基准数据集上取得了具有竞争力的性能,在ToxiGen上准确率达到90.97%,在Jigsaw数据集上准确率达到86.98%,可与规模超过其50倍的模型相媲美。这种高效性使其能够部署在资源受限的环境中,满足了对兼顾性能与计算效率的有效内容审核工具的需求。该模型架构包含4个Transformer编码器层,每层有2个注意力头,嵌入维度为64,前馈维度为128。Tiny-toxic-detector在公开和私有数据集上进行训练,展示了高效、任务特定模型在应对在线毒性问题方面的潜力。本文涵盖了模型架构、训练过程、性能基准和局限性,强调了其适用于社交媒体监控和内容审核等应用。通过取得与规模大得多的模型相当的结果,同时显著降低计算需求,Tiny-toxic-detector代表了向更可持续、可扩展的AI驱动内容审核解决方案迈进的进展。

英文摘要:

This paper presents Tiny-toxic-detector, a compact transformer-based model designed for toxic content detection. Despite having only 2.1 million parameters, Tiny-toxic-detector achieves competitive performance on benchmark datasets, with 90.97% accuracy on ToxiGen and 86.98% accuracy on the Jigsaw dataset, rivaling models over 50 times its size. This efficiency enables deployment in resource-constrained environments, addressing the need for effective content moderation tools that balance performance with computational efficiency. The model architecture features 4 transformer encoder layers, each with 2 attention heads, an embedding dimension of 64, and a feedforward dimension of 128. Trained on both public and private datasets, Tiny-toxic-detector demonstrates the potential of efficient, task-specific models for addressing online toxicity. The paper covers the model architecture, training process, performance benchmarks, and limitations, underscoring its suitability for applications such as social media monitoring and content moderation. By achieving results comparable to much larger models while significantly reducing computational demands, Tiny-toxic-detector represents progress toward more sustainable and scalable AI-driven content moderation solutions.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑