atlas

Research map / Large language models / Language, safety and interpretability

Efficient Toxic Content Detection by Bootstrapping and Distilling Large Language Models

AAAI 2024 · Large language models › Language, safety and interpretability · approach: LLM unlearning and benchmarks · 21 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes BD-LLM, which bootstraps LLMs with a confidence-gated, context-tree prompting method (DToT) and distills DToT rationales into a small FastChat-T5 student for toxic content detection.

Problem

Toxic content detection with supervised LMs has limited accuracy and cross-dataset transfer, while LLM prompting is hard to design and too costly to deploy in production.

Approach

Decision-Tree-of-Thought (DToT) iteratively re-prompts an LLM with more fine-grained toxicity-category context when its confidence is low, selecting the child context via rationale relevance; the extracted answers and rationales are then distilled into a 3B student LM that predicts both label and rationale.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace CoT prompting with DToT promptingChatGPT · Toxigenaccuracy: 82.71 -> 85.03Helped
Replace CoT prompting with DToT promptingChatGPT · SBICaccuracy: 68.50 -> 71.60Helped
Replace CoT prompting with DToT promptingChatGPT · DHateaccuracy: 65.20 -> 68.20Helped
Add few-shot demonstrations to DToT (ablation)ChatGPT · Toxigenaccuracy: 85.03 -> 86.03Helped
Add rationale augmentations to DToT with few-shot demonstrations (ablation)ChatGPT · Toxigenaccuracy: 86.03 -> 87.03Helped
Add few-shot demonstrations and rationale augmentations to DToT (ablation)ChatGPT · DHateaccuracy: 68.20 -> 69.20Mixed
Add DToT-extracted rationales during fine-tuning (ablation)FC-T5 with human labels · Toxigenaccuracy: 84.99 -> 87.53Helped
Use DToT-extracted rather than CoT-extracted rationales during fine-tuning (ablation)FC-T5 with human labels · Toxigenaccuracy: 87.31 -> 87.53Helped
Add DToT-extracted rationales during fine-tuning (ablation)FC-T5 with human labels · SBICaccuracy: 84.00 -> 85.10Helped
Use DToT-extracted rather than CoT-extracted rationales during fine-tuning (ablation)FC-T5 with human labels · DHateaccuracy: 86.20 -> 86.40Helped
Add DToT-extracted rationales during fine-tuning (ablation)FC-T5 fine-tuned on Toxigen · Toxigenaccuracy: 84.99 -> 87.53Helped
Add DToT-extracted rationales during fine-tuning (ablation)Flan-T5-XL 3B · Toxigenaccuracy: 85.54 -> 87.53Helped

Surprising findings

Limitations

Experimental setup

Datasets: Toxigen SBIC DHate Amazon internal toxic dataset

Benchmarks: Toxigen SBIC DHate Amazon internal dataset

Metrics: accuracy F1 AUC

Baselines: RoBERTa fine-tuned CoT prompting UniLC FC-T5 with labels only FC-T5 with CoT rationales

Builds on: Tree-of-Thought chain-of-thought UniLC Flan-T5 SCOTT

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy