TSE-IO: Two-Stage Extrapolative I/O Modeling and Tuning via Pairwise Learning
Collecting I/O data for parallel programs at large node scales is prohibitively expensive and has long been a major bottleneck for I/O performance modeling. Existing approaches typically depend on extensive training samples collected on large-scale systems, which incur high cost and generalize poorly across applications and environments. To address these challenges, we propose TSE-IO, a T wo- S tage framework for E xtrapolative I/O performance modeling and tuning. TSE-IO decomposes cross-scale performance prediction into two linked tasks: first, extrapolating application-level I/O behavior from small-scale measurements to the target scale. And second, predicting target-scale performance by jointly modeling the extrapolated behavior and tunable I/O-stack configurations. For the second stage, TSE-IO introduces a pairwise formulation that combines delta-feature matching with anchor-based inference to transfer configuration-induced performance patterns learned at small scales to unseen target scales. Experimental results show that, when trained solely on measurements collected from 1-16 nodes, TSE-IO extrapolates the I/O time of IOR, S3D-IO, Flash-IO, FLASH, and LAMMPS to 128 nodes (2,048 processes), achieving a MAPE below 20% for the majority of evaluated workloads. After tuning, five representative applications exhibit significant I/O performance improvements, achieving speedups ranging from 16.38 × to 65.38 ×, and from 1.00 × to 2.79 × compared with expert user settings across different workloads, thereby demonstrating the effectiveness and generality of TSE-IO.
Authors
- Xiaoshe Dong (ORCID: https://orcid.org/0000-0002-9003-2625)
- Ziheng Wang (ORCID: https://orcid.org/0000-0001-5064-2376)
- Qiang Wang (ORCID: https://orcid.org/0000-0001-9179-6611)
- Yifan Li (ORCID: https://orcid.org/0009-0005-4483-0318)
- LIU Xindong (ORCID: https://orcid.org/0009-0009-2188-6895)
- BingZhi Long (ORCID: https://orcid.org/0009-0003-1319-5169)
- Heng Liu (ORCID: https://orcid.org/0009-0000-5074-935X)
Institutions
- Xi'an Jiaotong University (CN)
Publication Details
- Journal
- ACM Transactions on Architecture and Code Optimization
- Published
- 2026-11-02
- DOI
- https://doi.org/10.1145/3860541
- Primary Topic
- Parallel Computing and Optimization Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.00