Constraint-Aware Hierarchical Search for Regulation-Driven Fine-Grained Classification
2026-07-12 • Artificial Intelligence
Artificial IntelligenceComputation and Language
AI summaryⓘ
The authors study tasks where items must be classified into detailed categories based on strict rules and hierarchies, like customs or export controls. Unlike normal text classification, these decisions depend on rules and exceptions, not just similarity. They create new datasets and propose a method that searches within the rules to find the right category while providing clear explanations. Their approach outperforms existing methods, especially for tricky cases with close categories and rule-based distinctions.
hierarchical classificationregulatory hierarchyrule-based classificationcustoms tariffexport controlfine-grained categorieshierarchical searchevidence retrievaltext classificationconstraint-aware
Authors
Siyu Wang, Wei Tan, Lulu Chen
Abstract
Tasks such as customs tariff classification, export control categorization, and standards-based equipment coding require assigning an input instance to a fine-grained class under an explicit regulatory hierarchy. Unlike standard text classification, the correct label in these tasks is not determined by semantic similarity alone, but by rule-defined boundaries, threshold conditions, exclusion clauses, definitions, and local exceptions. As a result, two highly similar inputs may require different labels, while a retrieved passage that appears relevant may still be inapplicable under the governing rules. Existing flat classifiers, hierarchical text classification methods, and retrieval-augmented LLM systems are not designed to jointly enforce hierarchical validity, rule consistency, and fine-grained boundary reasoning. In this paper, we formulate this setting as regulation-driven fine-grained hierarchical classification, where an external instance must be assigned to a fine-grained class through a valid path in a regulatory hierarchy and supported by auditable evidence. We construct four benchmark datasets from representative regulation-intensive scenarios and validate the annotations through an expert-in-the-loop process. We further propose a constraint-aware hierarchical search framework that converts regulatory documents into a searchable tree, retrieves only valid local candidate nodes, and uses structured regulatory fields with evidence snippets to guide each next-hop decision. Experiments show that our method achieves the best mean accuracy on all four datasets and provides interpretable decision paths, with the largest gains on cases involving fine-grained neighboring categories and rule-based boundary conditions.