Dynamic Tokenization via Reinforcement Patching: End-to-end Training and Zero-shot Transfer
Not on the map. Why?
Abstract (arXiv)
Efficiently aggregating spatial or temporal horizons to acquire compact representations has become a unifying principle in modern deep learning models, yet learning data-adaptive representations for long-horizon sequence data, especially continuous sequences like time series, remains an open challenge. While fixed-size patching has improved scalability and performance, discovering variable-sized, data-driven patches end-to-end often forces models to rely on soft discretization, specific backbones, or heuristic rules. In this work, we propose Reinforcement Patching (ReinPatch), the first framework to jointly optimize a sequence patching policy and its downstream sequence backbone model using reinforcement learning. By formulating patch boundary placement as a discrete decision process optimized via Group Relative Policy Gradient (GRPG), ReinPatch bypasses the need for continuous relaxations and performs dynamic patching policy optimization in a natural manner. Moreover, our method allows strict enforcement of a desired compression rate, freeing the downstream backbone to scale efficiently, and naturally supports multi-level hierarchical modeling. We evaluate ReinPatch on time-series forecasting datasets, where it demonstrates compelling performance compared to state-of-the-art data-driven patching strategies. Furthermore, our detached design allows the patching module to be extracted as a standalone foundation patcher, providing the community with visual and empirical insights into the segmentation behaviors preferred by a purely performance-driven neural patching strategy.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 25 of its 28 references that are in the map. First posted on arXiv 27 Mar 2026.
Builds on
25 of 28 references.
References · 25
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2015Neural Machine Translation of Rare Words with Subword Unitscited by 2,898
- 2016Asynchronous Methods for Deep Reinforcement Learningcited by 2,218
- 2020Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecastingcited by 2,141
- 2022A Time Series is Worth 64 Words: Long-term Forecasting with Transformerscited by 1,981
- 2021Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecastingcited by 1,654
- 2022Are Transformers Effective for Time Series Forecasting?cited by 1,400
- 2022FEDformer: Frequency Enhanced Decomposed Transformer for Long-term Series Forecastingcited by 1,200
13 more
- 2023A decoder-only foundation model for time-series forecastingcited by 650
- 2024Unified Training of Universal Time Series Forecasting Transformerscited by 588
- 2024MOMENT: A Family of Open Time-series Foundation Modelscited by 453
- 2024Timer: Generative Pre-trained Transformers Are Large Time Series Modelscited by 198
- 2024Byte Latent Transformer: Patches Scale Better Than Tokenscited by 137
- 2024Timer-XL: Long-Context Transformers for Unified Time Series Forecastingcited by 83
- 2022Efficient Transformers with Dynamic Token Poolingcited by 75
- 2025Dynamic Chunking for End-to-End Hierarchical Sequence Modelingcited by 65
- 2025Enhancing Time Series Forecasting through Selective Representation Spaces: A Patch Perspectivecited by 34
- 2024SpaceByte: Towards Deleting Tokenization from Large Language Modelingcited by 31
- 2025LightGTS: A Lightweight General Time Series Forecasting Modelcited by 28
- 2026TimeSqueeze: Dynamic Patching for Efficient Time Series Forecastingcited by 3
- 2025Entropy Guided Dynamic Patch Segmentation for Time Series Transformerscited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.