AutoDetect: Towards a Unified Framework for Automated Weakness Detection in Large Language Models

Jiale Cheng, Yida Lu, Xiaotao Gu, Pei Ke, Xiao Liu +4 more
6/24/2024
cs.CLcs.AIcs.LG

Official code: thu-coai/AutoDetect (47 stars, MIT, last commit 2024-06-25).

Abstract

Although Large Language Models (LLMs) are becoming increasingly powerful, they still exhibit significant but subtle weaknesses, such as mistakes in instruction-following or coding tasks. As these unexpected errors could lead to severe consequences in practical deployments, it is crucial to investigate the limitations within LLMs systematically. Traditional benchmarking approaches cannot thoroughly pinpoint specific model deficiencies, while manual inspections are costly and not scalable. In this paper, we introduce a unified framework, AutoDetect, to automatically expose weaknesses in LLMs across various tasks. Inspired by the educational assessment process that measures students' learning outcomes, AutoDetect consists of three LLM-powered agents: Examiner, Questioner, and Assessor. The collaboration among these three agents is designed to realize comprehensive and in-depth weakness identification. Our framework demonstrates significant success in uncovering flaws, with an identification success rate exceeding 30% in prominent models such as ChatGPT and Claude. More importantly, these identified weaknesses can guide specific model improvements, proving more effective than untargeted data augmentation methods like Self-Instruct. Our approach has led to substantial enhancements in popular LLMs, including the Llama series and Mistral-7b, boosting their performance by over 10% across several benchmarks. Code and data are publicly available at https://github.com/thu-coai/AutoDetect.

View on arXivView PDF

Code Implementations(1)

MIT

Official github repo for AutoDetect, an automated weakness detection framework for LLMs.

471Shell, PythonJun 24, 20242 years agoMIT

Similar Papers

Cite this paper

@article{cheng2024autodetect,
  title  = {AutoDetect: Towards a Unified Framework for Automated Weakness Detection in Large Language Models},
  author = {Jiale Cheng and Yida Lu and Xiaotao Gu and Pei Ke and Xiao Liu and Yuxiao Dong and Hongning Wang and Jie Tang and Minlie Huang},
  year   = {2024},
  eprint = {2406.16714},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2406.16714v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2406.16714.svg)](https://codeofpaper.com/paper/2406.16714)

Discussion