Breaking the Pre-Sampling Barrier: Activation-Informed Difficulty-Aware Self-Consistency

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yoon, Taewoong, Jeong, Geunyeong, Park, Geon, Yeom, Sihyeong, Kim, Harksoo
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911437173030912
author Yoon, Taewoong
Jeong, Geunyeong
Park, Geon
Yeom, Sihyeong
Kim, Harksoo
author_facet Yoon, Taewoong
Jeong, Geunyeong
Park, Geon
Yeom, Sihyeong
Kim, Harksoo
contents Self-Consistency (SC) is an effective decoding strategy that improves the reasoning performance of Large Language Models (LLMs) by generating multiple chain-of-thought reasoning paths and selecting the final answer via majority voting. However, it suffers from substantial inference costs because it requires a large number of samples. To mitigate this issue, Difficulty-Adaptive Self-Consistency (DSC) was proposed to reduce unnecessary token usage for easy problems by adjusting the number of samples according to problem difficulty. However, DSC requires additional model calls and pre-sampling to estimate difficulty, and this process is repeated when applying to each dataset, leading to significant computational overhead. In this work, we propose Activation-Informed Difficulty-Aware Self-Consistency (ACTSC) to address these limitations. ACTSC leverages internal difficulty signals reflected in the feed-forward network neuron activations to construct a lightweight difficulty estimation probe, without any additional token generation or model calls. The probe dynamically adjusts the number of samples for SC and can be applied to new datasets without requiring pre-sampling for difficulty estimation. To validate its effectiveness, we conduct experiments on five benchmarks. Experimental results show that ACTSC effectively reduces inference costs while maintaining accuracy relative to existing methods.
format Preprint
id arxiv_https___arxiv_org_abs_2602_09438
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Breaking the Pre-Sampling Barrier: Activation-Informed Difficulty-Aware Self-Consistency
Yoon, Taewoong
Jeong, Geunyeong
Park, Geon
Yeom, Sihyeong
Kim, Harksoo
Computation and Language
Self-Consistency (SC) is an effective decoding strategy that improves the reasoning performance of Large Language Models (LLMs) by generating multiple chain-of-thought reasoning paths and selecting the final answer via majority voting. However, it suffers from substantial inference costs because it requires a large number of samples. To mitigate this issue, Difficulty-Adaptive Self-Consistency (DSC) was proposed to reduce unnecessary token usage for easy problems by adjusting the number of samples according to problem difficulty. However, DSC requires additional model calls and pre-sampling to estimate difficulty, and this process is repeated when applying to each dataset, leading to significant computational overhead. In this work, we propose Activation-Informed Difficulty-Aware Self-Consistency (ACTSC) to address these limitations. ACTSC leverages internal difficulty signals reflected in the feed-forward network neuron activations to construct a lightweight difficulty estimation probe, without any additional token generation or model calls. The probe dynamically adjusts the number of samples for SC and can be applied to new datasets without requiring pre-sampling for difficulty estimation. To validate its effectiveness, we conduct experiments on five benchmarks. Experimental results show that ACTSC effectively reduces inference costs while maintaining accuracy relative to existing methods.
title Breaking the Pre-Sampling Barrier: Activation-Informed Difficulty-Aware Self-Consistency
topic Computation and Language
url https://arxiv.org/abs/2602.09438