Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Max-pooling Network Revisited: Analyzing the Role of Semantic Probability in Multiple Instance Learning for Hallucination Detection

About

Hallucination detection has become increasingly important for improving the reliability of large language models (LLMs). Recently, hybrid approaches such as HaMI, which combine semantic consistency with internal model states via Multiple Instance Learning (MIL), have achieved state-of-the-art performance. However, these methods incur substantial computational overhead due to repeated sampling and costly semantic similarity computations. In this work, we first provide a theoretical analysis of HaMI in terms of decision margins, revealing that scaling internal states with semantic consistency leads to an enlarged decision margin. Motivated by this insight, we revisit classical sentence classification models from a margin enlargement perspective, aggregating token-level features via max pooling and directly estimating sentence scores using a lightweight MLP. Without requiring semantic consistency computations, our approach achieves substantial efficiency improvements while maintaining competitive performance with state-of-the-art baselines through adaptive aggregation of internal feature representations. Code is available at https://github.com/FUJI1229/Hallucination_Detection.

Shota Fujikawa, Issei Sato• 2026

Related benchmarks

TaskDatasetResultRank
Hallucination DetectionTriviaQA--
621
Hallucination DetectionNQ
AUC0.889
154
Hallucination DetectionBioASQ
AUROC0.917
104
Hallucination DetectionSQuAD--
28
Hallucination DetectionSQuAD
Inference Throughput (Samples/sec)5.30e+3
12
Hallucination DetectionBioASQ
Inference Throughput (Samples/sec)5.35e+3
12
Hallucination DetectionNQ
Throughput (Samples/sec)4.99e+3
12
Showing 7 of 7 rows

Other info

Follow for update