Paper Reading AI Learner

Toward General Semantic Chunking: A Discriminative Framework for Ultra-Long Documents

2025-12-23 16:16:51
Kaifeng Wu, Junyan Wu, Qiang Liu, Jiarui Zhang, Wen Xu

Abstract

Long-document topic segmentation plays an important role in information retrieval and document understanding, yet existing methods still show clear shortcomings in ultra-long text settings. Traditional discriminative models are constrained by fixed windows and cannot model document-level semantics; generative large language models can output paragraph boundaries, but inference is expensive and long inputs are difficult to support. To address these issues, we propose a discriminative segmentation model based on Qwen3-0.6B. On top of the backbone network, we add a cross-window context fusion layer and a boundary classification head, and combine them with an overlapping sliding-window strategy. Our model supports single-pass inputs of up to 13k tokens and can be extended to ultra-long documents for paragraph boundary detection. To further enhance downstream retrieval efficiency, we derive a vector fusion method with scalar correction, which compresses the representation of ultra-long segments into a single vector without semantic loss. Experiments on the Wikipedia long-document topic segmentation dataset WIKI-727K show that, compared with three generative models based on Qwen2-0.5B released by Jina, our method achieves a better macro-averaged F1 and delivers two orders of magnitude faster inference, substantially improving the practicality and scalability of long-document processing.

Abstract (translated)

长文档主题分段在信息检索和文档理解中扮演着重要角色,但现有方法在处理超长文本时仍存在明显不足。传统判别模型受到固定窗口的限制,无法捕捉到文档级别的语义;而生成式的大语言模型能够输出段落边界,但在推理过程中成本高昂,并且难以支持长输入。为解决这些问题,我们提出了一种基于Qwen3-0.6B的判别分段模型。在基础网络之上,我们添加了一个跨窗口上下文融合层和一个边界分类头,并结合了重叠滑动窗口策略。我们的模型能够单次处理长达13K令牌的输入,并可扩展至超长文档进行段落边界的检测。 为了进一步提高下游检索效率,我们推导出了一种向量融合方法,该方法通过标量修正将超长片段的表示压缩为单一向量,同时保持语义不变。在维基百科长文档主题分段数据集WIKI-727K上的实验表明,与Jina基于Qwen2-0.5B发布的三种生成模型相比,我们的方法实现了更好的宏平均F1分数,并且推理速度快两个数量级,显著提高了处理长文档的实用性和可扩展性。

URL

https://arxiv.org/abs/2602.23370

PDF

https://arxiv.org/pdf/2602.23370.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot