New Benchmark Reveals Document Sensitivity Classification Challenges

Aleesha Zainab, Muhammad Ahmed Khalid, Faheem Ullah Khan, Asifullah Khan· August 19, 2026 View original

Key takeaways

  • Label leakage in training data inflates document sensitivity classification performance.
  • Strategic 16K is a new leakage-controlled benchmark dataset for this task.
  • BERT and ELECTRA show strong performance on clean, sensitive document data.
  • Classical models like TF-IDF with Logistic Regression offer a cost-effective alternative.

Who benefits

GovernmentLegalFinanceHealthcareCybersecurity

Summary

This paper introduces Strategic 16K, a new leakage-controlled dataset of diplomatic cables, to benchmark document sensitivity classification models. It highlights the pervasive issue of label leakage in training data and evaluates classical and transformer models, finding BERT performs best on the clean data.

Automatically classifying the sensitivity of organizational documents is a critical task, with misclassifications potentially leading to severe regulatory or security issues. While AI offers a scalable solution, its reliability hinges on the quality of training data. A significant, often overlooked problem is "label leakage," where models exploit superficial markers in documents rather than learning genuine content-based sensitivity, leading to inflated performance metrics. To address this, researchers have developed Strategic 16K, a meticulously curated dataset of 16,000 diplomatic cables from the WikiLeaks Public Library of US Diplomacy (PlusD). This dataset is specifically designed to be leakage-controlled, meaning it has undergone an extensive protocol to remove residual classification markers that could mislead models. The study benchmarks six different model architectures, including classical machine learning methods and transformer-based approaches, on this clean dataset. BERT emerged as the top performer, followed closely by ELECTRA. Among classical models, TF-IDF with Logistic Regression showed strong performance at a much lower computational cost. This work provides the first fully reproducible, leakage-controlled benchmark for document sensitivity classification using WikiLeaks PlusD data.

Why it matters

Professionals dealing with sensitive information can now better assess the true performance of AI models for document classification, avoiding over-reliance on models trained on compromised data. This improves data security and compliance.

How to implement this in your domain

  1. 1Review existing document classification systems for potential label leakage issues in their training data.
  2. 2Adopt rigorous data cleaning protocols, similar to the extended leakage removal described, for sensitive document datasets.
  3. 3Benchmark current classification models against a leakage-controlled dataset to get a realistic performance assessment.
  4. 4Consider transformer-based models like BERT for high-accuracy sensitivity classification, especially for complex documents.
  5. 5Evaluate the trade-off between accuracy and computational cost, potentially using classical models for less critical or resource-constrained applications.

Original post by Aleesha Zainab, Muhammad Ahmed Khalid, Faheem Ullah Khan, Asifullah Khan

"arXiv:2608.16928v1 Announce Type: new Abstract: Automatic sensitivity classification of organizational documents is a critical yet underserved problem, where the consequences of misclassification range from regulatory violations to security breaches. While AI-based approaches off…"

View on X

Originally posted by Aleesha Zainab, Muhammad Ahmed Khalid, Faheem Ullah Khan, Asifullah Khan on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses

More in AI Engineering & DevTools