Leakage-Safe and Scheduler-Aware Machine Learning for Grid Job Runtime Prediction

arXiv cs.LG Papers

Summary

This paper presents a leakage-safe and scheduler-aware machine learning framework for predicting grid job runtime, using CatBoost with temporal validation to enhance scheduling efficiency in distributed computing environments.

arXiv:2609.13701v1 Announce Type: new Abstract: Accurate job runtime prediction can improve scheduling-aware resource management in grid and distributed computing environments, but prediction models must be evaluated under realistic deployment constraints. This paper revisits CPU burst time prediction on the GWA-T-4 AuverGrid workload trace and reformulates it as leakage-safe pre-execution job runtime prediction. We define the target as job-level runtime, use only submission-time attributes, exclude post-execution variables, and evaluate models under temporal and cold-start settings rather than relying only on random cross-validation. We compare standard regressors, chronological historical baselines, categorical encoding strategies, and CatBoost with native categorical handling. We further add temporal hyperparameter tuning, runtime predictability analysis, feature ablation, error analysis by job length, and a minimal scheduling simulation. After temporal-validation tuning, CatBoost achieves the strongest deployment-oriented result with R^2=0.239, MAE=27,019, RMSE=46,587, and LogMAE=2.646 on the held-out temporal test set. A single-server simulation over all 69,523 held-out temporal test jobs shows that prediction-informed SJF reduces average waiting time by 50.92% relative to FCFS. The results show that random-split evaluation overestimates performance, categorical-native boosting improves temporal generalization, and long-job underestimation remains a scheduler-relevant challenge.
Original Article
View Cached Full Text

Cached at: 09/15/26, 08:49 AM

# Leakage-Safe and Scheduler-Aware Machine Learning for Grid Job Runtime Prediction
Source: [https://arxiv.org/html/2609.13701](https://arxiv.org/html/2609.13701)
Ashfaq Ali ShafinAffiliation:Augustana College, USA Florida International University, USA Email: shafinashfaqali21@gmail\.comKhandaker Mamun Ahmed

###### Abstract

Accurate job runtime prediction can improve scheduling\-aware resource management in grid and distributed computing environments, but prediction models must be evaluated under realistic deployment constraints\. This paper revisits CPU burst time prediction on the GWA\-T\-4 AuverGrid workload trace and reformulates it as leakage\-safe pre\-execution job runtime prediction\. We define the target as job\-level runtime, use only submission\-time attributes, exclude post\-execution variables, and evaluate models under temporal and cold\-start settings rather than relying only on random cross\-validation\. We compare standard regressors, chronological historical baselines, categorical encoding strategies, and CatBoost with native categorical handling\. We further add temporal hyperparameter tuning, runtime predictability analysis, feature ablation, error analysis by job length, and a minimal scheduling simulation\. After temporal\-validation tuning, CatBoost achieves the strongest deployment\-oriented result withR2=0\.239R^\{2\}=0\.239, MAE=27,019=27\{,\}019, RMSE=46,587=46\{,\}587, and LogMAE=2\.646=2\.646on the held\-out temporal test set\. A single\-server simulation over all 69,523 held\-out temporal test jobs shows that prediction\-informed SJF reduces average waiting time by 50\.92% relative to FCFS\. The results show that random\-split evaluation overestimates performance, categorical\-native boosting improves temporal generalization, and long\-job underestimation remains a scheduler\-relevant challenge\.

###### Index Terms:

grid workload prediction, runtime prediction, CPU burst time, leakage\-safe evaluation, deployment\-aware evaluation, CatBoost, scheduling simulation, machine learning, distributed systems

## IIntroduction

Process and job scheduling are fundamental functions in operating systems, grid platforms, and distributed computing environments\. In a multitasking operating system, a process scheduler determines which process should be assigned to the CPU\. Common scheduling algorithms include First Come First Serve \(FCFS\), Round\-Robin \(RR\), Priority Scheduling, Shortest Job First \(SJF\), and Shortest Remaining Time First \(SRTF\)\[[1](https://arxiv.org/html/2609.13701#bib.bib10)\]\. In grid and high\-performance computing environments, scheduling decisions affect waiting time, turnaround time, slowdown, fairness, resource utilization, and throughput\.

A CPU burst refers to the interval during which a process executes on the CPU before it blocks, terminates, or leaves the running state\[[2](https://arxiv.org/html/2609.13701#bib.bib8)\]\. In production grid workload traces, however, the available target is usually job\-level runtime rather than fine\-grained CPU bursts between I/O operations\. Therefore, this paper uses the term*CPU burst time prediction*in the scheduling sense, but defines the measurable prediction target precisely as the job\-level runtime recorded in the GWA\-T\-4 AuverGrid trace\. This definition avoids ambiguity between operating\-system\-level CPU bursts and grid\-workload job durations\.

Reliable runtime estimates are useful for multiple scheduling policies\. SJF and SRTF prioritize jobs with shorter expected execution times\. Backfilling and reservation\-based schedulers use runtime estimates to decide whether a job can be executed without delaying a reserved higher\-priority job\. Adaptive variants of Round\-Robin scheduling may also use runtime distributions to determine time quanta\. In practice, however, the actual runtime of a newly submitted job is unknown before execution\. A scheduler must estimate runtime using only information available at or before job submission\.

This requirement creates a methodological challenge\. Workload traces contain both pre\-execution variables and post\-execution outcomes\. If post\-execution variables such as observed runtime, waiting time, job status, used memory, CPU time used, partition assignment, or last run site are used as input features, the prediction task becomes invalid for pre\-scheduling deployment\. Such variables are unavailable when the scheduler must make its decision and can cause target leakage\. A model may then appear highly accurate because it indirectly observes the target or future execution outcomes\.

Machine learning offers a promising approach for runtime prediction because supervised models can learn relationships among job submission attributes, requested resources, user behavior, group membership, queue assignment, site information, and executable identifiers\. Prior work has applied learning\-based approaches to runtime prediction and scheduling\-related decisions\[[3](https://arxiv.org/html/2609.13701#bib.bib5),[4](https://arxiv.org/html/2609.13701#bib.bib6),[5](https://arxiv.org/html/2609.13701#bib.bib4),[6](https://arxiv.org/html/2609.13701#bib.bib11)\]\. However, for machine learning to be useful in scheduling\-aware systems, evaluation must be leakage\-safe, time\-aware, and connected to scheduling behavior rather than reported only as standalone regression accuracy\.

This paper revises and strengthens CPU burst time prediction using the GWA\-T\-4 AuverGrid workload trace\[[7](https://arxiv.org/html/2609.13701#bib.bib3)\]\. The central contribution is not a new regression algorithm, but a rigorous deployment\-aware reevaluation of grid runtime prediction under realistic information constraints\. We define a strict pre\-execution prediction task, exclude all attributes not known at submission time, fit preprocessing only on training partitions, evaluate random and deployment\-oriented splits, add chronological\-safe historical baselines, compare categorical encoding strategies, tune CatBoost using a temporal validation split, analyze runtime predictability, explain feature ablation results, and add a minimal scheduling simulation\.

The study addresses the following research questions:

- •RQ1:How accurately can machine learning models predict grid job runtime using only attributes available at job submission time?
- •RQ2:How do learning\-based models compare with chronological historical baselines that use only past jobs relative to each test job?
- •RQ3:How do categorical encoding and native categorical modeling affect temporal generalization?
- •RQ4:Which pre\-submit feature groups contribute most to temporal prediction performance?
- •RQ5:How do prediction errors differ across short, medium, long, very long, and extreme\-duration jobs?
- •RQ6:Can prediction\-informed scheduling improve waiting time in a minimal scheduling simulation?

The main contributions are:

- •We clarify the prediction target and define CPU burst time prediction as job\-level runtime prediction on the AuverGrid trace\.
- •We introduce a leakage\-safe feature policy that uses only pre\-submit attributes and excludes post\-execution outcomes\.
- •We evaluate chronological\-safe historical baselines that prevent future information from influencing baseline predictions\.
- •We compare one\-hot, ordinal, target\-mean, and native categorical modeling strategies, including temporal\-validation\-tuned CatBoost\.
- •We add runtime predictability analysis, feature ablation, permutation importance, completed\-job filtering, and error\-bias analysis by job\-length group\.
- •We provide a minimal scheduling simulation showing that prediction\-informed SJF can reduce waiting time relative to FCFS in a simplified non\-preemptive setting\.

## IIBackground and Related Work

### II\-AClassical Runtime and Burst\-Time Estimation

Several scheduling algorithms require estimates of future burst time or job duration\. A classical approach is exponential averaging, where the next predicted burst is computed from the most recent observed burst and the previous prediction:

τn\+1=α​Tn\+\(1−α\)​τn,\\tau\_\{n\+1\}=\\alpha T\_\{n\}\+\(1\-\\alpha\)\\tau\_\{n\},\(1\)
whereTnT\_\{n\}is the most recent observed runtime,τn\\tau\_\{n\}is the previous estimate, andα\\alphacontrols the influence of recent history\. This formulation is common in operating\-systems scheduling discussions\[[1](https://arxiv.org/html/2609.13701#bib.bib10),[8](https://arxiv.org/html/2609.13701#bib.bib9)\]\. Its advantage is simplicity and low computational overhead, but it uses limited information and may be insufficient for heterogeneous grid workloads\.

Optimization\-based and rule\-based approaches have also been explored\. Maheshwari et al\. proposed a scheduling approach using Linear Programming Models Dual Simplex Algorithm \(LPMDSA\), where burst time, waiting time, and turnaround time are represented mathematically and optimized using the dual simplex method\[[9](https://arxiv.org/html/2609.13701#bib.bib7)\]\. Pourali et al\. proposed an intelligent fuzzy system to estimate CPU burst time from previous process behavior\[[10](https://arxiv.org/html/2609.13701#bib.bib2)\]\. These studies demonstrate that historical execution behavior can be useful, but they often require manually designed rules or problem\-specific assumptions\.

### II\-BMachine Learning for Runtime Prediction

Machine learning has been widely explored for runtime prediction and scheduling\-related decisions\. Smith et al\. proposed predicting parallel application runtimes by identifying similar past applications and using search methods to determine which application characteristics define similarity\[[5](https://arxiv.org/html/2609.13701#bib.bib4)\]\. Negi et al\. applied machine learning to CPU scheduling in Linux, aiming to reduce process turnaround time by learning CPU time\-slice utilization behavior\[[3](https://arxiv.org/html/2609.13701#bib.bib5)\]\. Shulga et al\. studied learned CPU/GPU execution selection in heterogeneous environments\[[4](https://arxiv.org/html/2609.13701#bib.bib6)\]\. Helmy et al\. studied CPU burst time estimation on the GWA\-T\-4 AuverGrid dataset using machine learning\[[6](https://arxiv.org/html/2609.13701#bib.bib11)\]\. Our work revisits this direction with stricter feature availability, chronological baselines, categorical\-native modeling, temporal tuning, and scheduler\-level validation\.

### II\-CScheduling Relevance and Evaluation Realism

Recent HPC runtime\-prediction work emphasizes that prediction should be connected to scheduling decisions rather than evaluated only as a standalone regression problem\. Duration\-Informed Workload Scheduler \(DIWS\) integrates machine\-learning runtime estimates into an HPC scheduling policy and evaluates the scheduling effect using simulation\[[11](https://arxiv.org/html/2609.13701#bib.bib12)\]\. RLBackfilling shows that better runtime prediction alone may not always yield better backfilling decisions because there is a trade\-off between prediction accuracy and backfilling opportunities\[[12](https://arxiv.org/html/2609.13701#bib.bib13)\]\. Best\-practice studies for HPC runtime prediction emphasize careful methodological design and operationally meaningful evaluation\[[13](https://arxiv.org/html/2609.13701#bib.bib14)\]\. Digital\-twin\-based scheduling frameworks further show that prediction and simulation can be combined to evaluate adaptive scheduling policies before deployment\[[14](https://arxiv.org/html/2609.13701#bib.bib15),[15](https://arxiv.org/html/2609.13701#bib.bib16)\]\. These works motivate our deployment\-aware prediction benchmark and our minimal scheduler\-level validation\.

## IIIDataset and Feature Policy

TABLE I:Dataset statistics before and after filtering\.TABLE II:Leakage\-safe feature policy for runtime prediction\.### III\-AGWA\-T\-4 AuverGrid Workload Trace

This study uses the GWA\-T\-4 AuverGrid workload dataset\[[7](https://arxiv.org/html/2609.13701#bib.bib3),[16](https://arxiv.org/html/2609.13701#bib.bib1)\]\(see Table[I](https://arxiv.org/html/2609.13701#S3.T1)\)\. AuverGrid is a production grid platform composed of multiple clusters located in the Auvergne region of France\. The trace contains job submission and execution records, including submission time, requested resources, user and group identifiers, executable identifiers, queue information, runtime, and post\-execution outcomes\.

The original dataset contains 404,176 jobs and 29 columns\. After filtering invalid runtime and submission\-time records and applying the leakage\-safe feature policy, the main experimental dataset contains 347,611 jobs\.

### III\-BTarget Definition

The prediction target is the job\-level runtime recorded in the AuverGrid trace\. We denote this value asyiy\_\{i\}for jobii\. The trace does not provide fine\-grained CPU bursts between I/O operations\. Therefore, throughout the experimental sections, CPU burst time refers to the runtime that a scheduler would like to estimate before job execution\.

### III\-CLeakage\-Safe Feature Policy

Table[II](https://arxiv.org/html/2609.13701#S3.T2)lists the revised feature policy\. Only pre\-submit fields are eligible as features\. Post\-execution variables are excluded because they are not available when the scheduler must make a decision\.

The final feature set contains 11 numeric features and 5 categorical features\. The numeric features are SubmitTime, SubmitTimeSinceStart, cyclic hour/day/week features, ReqNProcs, ReqTime, and ReqMemory\. The categorical features are UserID, GroupID, ExecutableID, QueueID, and OrigSiteID\.

### III\-DSubmitTime Feature Engineering

Raw submission time can be difficult for models to use directly\. We therefore derive temporal features that capture both monotonic trend and periodic workload structure\. Letsis\_\{i\}be the submission time for jobiiands0s\_\{0\}be the earliest submission time in the retained trace\. We define:

Δ​si=si−s0\.\\Delta s\_\{i\}=s\_\{i\}\-s\_\{0\}\.\(2\)
Hour\-of\-day, day\-of\-week, and week\-of\-year signals are represented using sine and cosine transformations:

sin⁡\(2​π​z/P\),cos⁡\(2​π​z/P\),\\sin\(2\\pi z/P\),\\quad\\cos\(2\\pi z/P\),\(3\)
wherezzis the corresponding hour, day, or week index andPPis the period length\. This representation avoids imposing artificial discontinuities between adjacent times, such as hour 23 and hour 0\.

## IVMethodology

### IV\-APrediction Task

We formulate grid job runtime prediction as supervised regression\. For each jobii, the input vector𝐱i\\mathbf\{x\}\_\{i\}contains only pre\-submit attributes, and the targetyiy\_\{i\}is observed runtime\. The goal is to learnf⁡\(𝐱i\)f\(\\mathbf\{x\}\_\{i\}\)that predictsy^i\\hat\{y\}\_\{i\}before the job executes\.

### IV\-BModels and Encoding Strategies

We evaluate linear, distance\-based, tree\-based, and boosting models\. The model set includes Linear Regression, Linear SVR, K\-NN, Decision Tree, Random Forest, Gradient Boosting, XGBoost, and CatBoost\. Because categorical identifiers are central to workload prediction, we compare one\-hot encoding, ordinal encoding, target\-mean encoding, and CatBoost’s native categorical handling\. K\-NN is evaluated with one\-hot encoding because ordinal codes impose artificial distances between unordered identifiers\.

CatBoost is included because it natively handles categorical variables using ordered target statistics and can reduce the need for high\-dimensional one\-hot encodings\. This is particularly relevant for workload traces where user and executable identifiers carry historical behavioral information\.

TABLE III:Runtime distribution and predictability analysis\. Entropy is estimated onlog⁡\(1\+RunTime\)\\log\(1\+\\text\{RunTime\}\)\.
### IV\-CChronological Historical Baselines

Historical baselines can accidentally use future information under random splits if they summarize all training records without respecting each test job’s submission time\. To avoid this issue, we implement chronological\-safe baselines\. For a test job submitted at timesis\_\{i\}, user, group, and executable baselines use only training jobs from the same entity with submission time less thansis\_\{i\}\. If no such history exists, the predictor falls back to the global training median\. This is applied to median and exponential\-moving\-average baselines\.

### IV\-DEvaluation Protocol

The revised evaluation includes random, temporal, user cold\-start, and executable cold\-start splits\. The main deployment\-oriented split is temporal 80/20: the earliest 80% of jobs are used for training and the latest 20% for testing\. Hyperparameter tuning uses an inner temporal validation split inside the training period and never uses the test partition for model selection\.

### IV\-ECatBoost Temporal Tuning

To ensure reproducibility, we tuned the CatBoost model using the inner temporal validation split\. The hyperparameter search space included tree depths of4,6,8\{4,6,8\}, iteration counts of200,300,500\{200,300,500\}, learning rates of0\.03,0\.05,0\.10\{0\.03,0\.05,0\.10\}, andL2L\_\{2\}leaf regularization values of1,3,5\{1,3,5\}\. The selected configuration used a tree depth of 8,500 iterations, a learning rate of 0\.03, and anL2L\_\{2\}leaf regularization value of 5\. The model was subsequently refit using this configuration on the full temporal training set and evaluated once on the held\-out temporal test set\.

### IV\-FRuntime Predictability Analysis

To contextualize low temporalR2R^\{2\}values, we analyze the intrinsic predictability of the workload\. Because runtime is long\-tailed, we estimate histogram entropy onlog⁡\(1\+RunTime\)\\log\(1\+\\text\{RunTime\}\)\. We report entropy in bits and normalized entropy relative to the maximum entropy of the nonempty histogram bins\. This analysis does not prove an irreducible error bound, but it provides a descriptive measure of runtime uncertainty\.

### IV\-GMinimal Scheduling Simulation

To connect prediction quality to scheduling behavior, we implement a minimal single\-server non\-preemptive scheduling simulation on the full held\-out temporal test set\. The simulation contains all 69,523 jobs from the latest 20% of the trace and compares three policies: FCFS, prediction\-informed SJF using tuned CatBoost runtime estimates, and oracle SJF using true runtimes\. Oracle SJF is not deployable and is included only as an upper bound\. The simulation is intentionally simplified and should not be interpreted as a full HPC backfilling simulator\. Its purpose is to provide a scheduler\-level evaluation showing that prediction errors can affect waiting time and turnaround time under the complete held\-out temporal workload\.

### IV\-HMetrics

We evaluate prediction performance using the coefficient of determination \(R2R^\{2\}\), mean absolute error \(MAE\), and root mean squared error \(RMSE\)\. Because grid runtimes exhibit a long\-tailed distribution, we additionally report the mean absolute error on the log\-transformed runtimes \(Log\-MAE\)\. The corresponding equations for these evaluation metrics are provided below\.

R2​\(y,y^\)=1−∑i=1n\(yi−y^i\)2∑i=1n\(yi−y¯\)2R^\{2\}\(y,\\hat\{y\}\)=1\-\\frac\{\\sum\_\{i=1\}^\{n\}\(y\_\{i\}\-\\hat\{y\}\_\{i\}\)^\{2\}\}\{\\sum\_\{i=1\}^\{n\}\(y\_\{i\}\-\\bar\{y\}\)^\{2\}\}\(4\)
M​A​E=1n​∑i=1n\|yi−y^i\|MAE=\\frac\{1\}\{n\}\\sum\_\{i=1\}^\{n\}\|y\_\{i\}\-\\hat\{y\}\_\{i\}\|\(5\)
R​M​S​E=1n​∑i=1n\(yi−y^i\)2RMSE=\\sqrt\{\\frac\{1\}\{n\}\\sum\_\{i=1\}^\{n\}\(y\_\{i\}\-\\hat\{y\}\_\{i\}\)^\{2\}\}\(6\)
L​o​g​M​A​E=1n​∑i=1n\|log⁡\(1\+yi\)−log⁡\(1\+y^i\)\|\.LogMAE=\\frac\{1\}\{n\}\\sum\_\{i=1\}^\{n\}\|\\log\(1\+y\_\{i\}\)\-\\log\(1\+\\hat\{y\}\_\{i\}\)\|\.\(7\)
TABLE IV:Tuned CatBoost performance on the temporal 80/20 split\.For scheduling simulation, we report average waiting time, median waiting time, average turnaround time, and percentage improvement in average waiting time relative to FCFS\.

## VExperimental Results

### V\-ARuntime Distribution and Predictability

Table[III](https://arxiv.org/html/2609.13701#S4.T3)summarizes the runtime distribution\. The retained workload is highly skewed: the median runtime is 2,658 seconds, while the mean is 25,802 seconds and the maximum observed runtime is 1,575,814 seconds\. The 95th and 99th percentiles are 97,796 and 173,200 seconds, respectively\. The normalized entropy oflog⁡\(1\+RunTime\)\\log\(1\+\\text\{RunTime\}\)is 0\.921, indicating substantial uncertainty even after log transformation\. This supports interpreting modest temporalR2R^\{2\}values in the context of a highly long\-tailed and nonstationary production workload\.

TABLE V:Best results across evaluation settings\. Random splitting is optimistic; temporal and cold\-start splits better reflect deployment\.TABLE VI:Chronological\-safe historical baselines\. For each test job, entity history uses only training jobs submitted earlier than the test job\.TABLE VII:Temporal 80/20 performance under categorical encoding strategies\. CatBoost uses native categorical handling and temporal\-validation\-selected hyperparameters\.
### V\-BMain Temporal Result

Table[IV](https://arxiv.org/html/2609.13701#S4.T4)reports the tuned CatBoost result on the temporal 80/20 split\. Hyperparameters are selected using only the inner temporal validation split\. The tuned CatBoost model achievesR2=0\.239R^\{2\}=0\.239, MAE=27,019=27\{,\}019, RMSE=46,587=46\{,\}587, and LogMAE=2\.646=2\.646on 69,523 held\-out later jobs\. The model trains in 17\.18 seconds and predicts the test set in 0\.10 seconds\.

TABLE VIII:CatBoost feature ablation on the temporal 80/20 split\.
### V\-CRandom, Temporal, and Cold\-Start Summary

Table[V](https://arxiv.org/html/2609.13701#S5.T5)summarizes the strongest results across evaluation settings\. Random splitting gives the strongest apparent performance, while temporal and cold\-start settings are substantially more difficult\. The tuned CatBoost temporal result is lower than the earlier untuned CatBoost value but is more reproducible because hyperparameters are selected without using the held\-out temporal test set\.

### V\-DChronological\-Safe Historical Baselines

Table[VI](https://arxiv.org/html/2609.13701#S5.T6)reports chronological\-safe historical baselines\. Under random 80/20 splitting, user\-level exponential moving average performs strongly withR2=0\.638R^\{2\}=0\.638and MAE=8,847=8\{,\}847\. This shows that user history is highly informative when similar user behavior is present in both training and test data\. Under temporal splitting, however, chronological baselines are weaker\. User median achievesR2=0\.064R^\{2\}=0\.064, while several other entity\-level baselines have negativeR2R^\{2\}values\. These results confirm that chronological consistency matters and that historical repetition alone does not fully solve future\-period generalization\.

### V\-ECategorical Encoding and CatBoost

Table[VII](https://arxiv.org/html/2609.13701#S5.T7)compares categorical encoding strategies on the temporal split\. Target\-mean encoding improves several conventional models compared with one\-hot and ordinal encoding\. Gradient Boosting with target encoding obtains the best conventional encoding\-comparison result withR2=0\.107R^\{2\}=0\.107\. Tuned CatBoost with native categorical handling performs substantially better, reachingR2=0\.239R^\{2\}=0\.239\.

This result supports the use of categorical\-native boosting for workload traces with high\-value user, group, and executable identifiers\. It also shows that ordinal encoding can be misleading because it imposes arbitrary numeric order on unordered identifiers\.

### V\-FFeature Ablation

The earlier Random Forest ablation produced the counterintuitive result that the all\-feature configuration performed worse than the identifier\-only configuration\. To examine whether this behavior was model\-specific, Table[VIII](https://arxiv.org/html/2609.13701#S5.T8)repeats the ablation using the tuned CatBoost model\. The all\-feature configuration achieves the strongest performance withR2=0\.239R^\{2\}=0\.239, compared withR2=0\.126R^\{2\}=0\.126for the identifier\-only configuration, while the requested\-resource\-only configuration yields a negativeR2R^\{2\}\. These results indicate that the earlier Random Forest degradation does not imply that the additional features are uninformative\. Instead, it likely reflects the difficulty of modeling sparse, high\-cardinality identifiers and their interactions with temporal and resource\-related features using one\-hot\-encoded tree ensembles\. CatBoost’s native categorical handling more effectively integrates identifier, requested\-resource, and temporal information\.

### V\-GCompleted\-Only Label Quality Analysis

The trace contains 336,085 completed jobs, representing 96\.68% of the filtered dataset\. Status values 0 and 5 account for the remaining 3\.32%\. Table[IX](https://arxiv.org/html/2609.13701#S5.T9)evaluates temporal prediction using only completed jobs\. The results remain low, although Gradient Boosting achieves the best completed\-onlyR2=0\.065R^\{2\}=0\.065\. This indicates that label\-quality filtering alone does not explain the temporal generalization challenge\.

TABLE IX:Completed\-only temporal prediction results\.
### V\-HPermutation Importance

Permutation importance for the temporal Random Forest model identifies UserID, GroupID, and ReqTime as the most influential features\. Table[X](https://arxiv.org/html/2609.13701#S5.T10)reports the highest\-importance features\. This confirms that repeated user and group behavior remains central to prediction, while requested time contributes additional signal\.

TABLE X:Permutation importance for Random Forest on temporal split\.
### V\-IError Analysis by Job Length

Table[XI](https://arxiv.org/html/2609.13701#S5.T11)reports error bias for tuned CatBoost across job\-length groups\. The model overestimates short and medium jobs but underestimates very long and extreme jobs\. For extreme jobs, the underestimation rate is 99\.97%, with a mean error of−105,448\-105\{,\}448seconds\. This is important for scheduling because underestimating long jobs can be more harmful than overestimating short jobs, especially in reservation\-based and backfilling schedulers\.

TABLE XI:Tuned CatBoost error analysis by job\-length group\. Positive error indicates overestimation; negative error indicates underestimation\.
### V\-JMinimal Scheduling Simulation

Table[XII](https://arxiv.org/html/2609.13701#S5.T12)reports the minimal single\-server scheduling simulation on all 69,523 held\-out temporal test jobs\. Predicted SJF uses tuned CatBoost runtime estimates\. Oracle SJF uses true runtimes and serves only as an upper bound\. Predicted SJF reduces average waiting time by 50\.92% relative to FCFS, while oracle SJF reduces average waiting time by 75\.83%\.

TABLE XII:Minimal single\-server scheduling simulation on all 69,523 held\-out temporal test jobs\. Predicted SJF uses tuned CatBoost runtime estimates\. Oracle SJF uses true runtimes and serves only as an upper bound\.This simulation does not replace a full production backfilling simulator\. It ignores multi\-node placement, queue priorities, reservations, resource fragmentation, and backfilling constraints\. However, because it is run on the complete held\-out temporal test set, it provides stronger evidence than a sampled sanity check: even moderate temporal prediction accuracy can produce meaningful waiting\-time improvements in a simplified non\-preemptive setting\. The large remaining gap between predicted SJF and oracle SJF also shows that prediction quality and long\-job underestimation remain important practical limitations\.

## VIDiscussion

### VI\-AWhat the Revised Results Show

The revised results change the interpretation of the study\. High random\-split scores should not be treated as deployment\-valid evidence\. Random splitting benefits from repeated users, groups, and applications appearing in both training and testing\. Temporal and cold\-start settings are more difficult and reveal the real generalization challenge\.

The tuned CatBoost result shows that categorical\-native boosting can recover meaningful temporal performance from pre\-submit features, reachingR2=0\.239R^\{2\}=0\.239on the held\-out temporal period\. The runtime entropy analysis helps explain why temporal performance remains modest: the workload is highly long\-tailed and uncertain, with normalized log\-runtime entropy 0\.921\. Therefore, the paper should not claim that runtime prediction is solved\. Instead, it should claim that leakage\-safe, categorical\-native modeling provides a more realistic and reproducible benchmark for deployment\-oriented prediction\.

### VI\-BWhy CatBoost Helps

Workload traces contain repeated behavioral patterns at the user, group, and executable levels\. One\-hot encoding can represent these categories but creates high\-dimensional sparse features\. Ordinal encoding imposes arbitrary order and can distort model behavior\. Target encoding can be useful but must be applied carefully to avoid leakage\. CatBoost provides an effective compromise because it handles categorical variables natively and uses ordered statistics\. The ablation results show that CatBoost benefits from combining identifiers, requested resources, and temporal features, unlike the earlier Random Forest one\-hot ablation\.

### VI\-CScheduling Implications

Prediction metrics do not automatically imply scheduling improvement\. Schedulers may be more sensitive to underestimation than overestimation because underestimating a long job can violate a backfilling reservation\. The minimal scheduling simulation over all held\-out temporal test jobs provides evidence that prediction\-informed scheduling can reduce waiting time relative to FCFS in a simplified single\-server setting, but it also confirms that a full scheduler\-integrated evaluation is needed\. The next step is to evaluate prediction\-informed EASY backfilling, conservative backfilling, SJF, and hybrid policies under realistic resource constraints\.

### VI\-DLong\-Job Underestimation

The error analysis shows that tuned CatBoost systematically underestimates very long and extreme jobs\. This failure mode is scheduler\-relevant\. Underestimation can increase reservation violations and delay other jobs if a scheduler assumes that a long job will finish earlier than it actually does\. This motivates asymmetric loss functions, quantile prediction, conformal prediction intervals, and conservative runtime estimates for backfilling\.

## VIIThreats to Validity

Several threats to validity remain\. First, the study uses one grid workload trace, so results may not generalize to all grid, cloud, or HPC environments\. Second, the trace provides job\-level runtime rather than fine\-grained CPU bursts between I/O operations\. Third, categorical identifiers are useful when histories exist but create cold\-start challenges\. Fourth, temporal validation does not fully eliminate distribution shift, as validation and test periods may differ\. Fifth, although the scheduling simulation uses the complete held\-out temporal test set, it is intentionally simplified and does not model a full production HPC scheduler, resource fragmentation, priorities, reservations, or EASY backfilling\. Sixth, entropy analysis describes distributional uncertainty but does not provide a formal irreducible\-error bound\.

## VIIIFuture Work

Future work should integrate the predictors into a full scheduling simulator and measure waiting time, turnaround time, bounded slowdown, utilization, fairness, and reservation violations\. We will evaluate prediction\-informed SJF, EASY backfilling, conservative backfilling, and hybrid policies\. We will also explore quantile regression, conformal prediction intervals, asymmetric loss functions that penalize long\-job underestimation, online learning for user and executable histories, and cross\-trace evaluation on additional workload traces\. A digital\-twin\-style evaluation framework could further support repeated what\-if comparisons of scheduling policies before deployment\.

## IXConclusion

This paper reformulated CPU burst time prediction for grid workload scheduling as leakage\-safe pre\-execution job runtime prediction\. The revised task defines the target as job\-level runtime in the AuverGrid trace and excludes all post\-execution variables from the feature set\. The paper adds chronological\-safe baselines, temporal feature engineering, categorical encoding comparisons, temporal\-validation\-tuned CatBoost, runtime predictability analysis, completed\-job filtering, feature ablation, permutation importance, error\-bias analysis, and a minimal scheduling simulation\. The results show that random splitting substantially overestimates deployment performance\. Tuned CatBoost improves temporal prediction under realistic feature constraints, while long\-job underestimation remains a major scheduling concern\. A minimal scheduling simulation over the complete held\-out temporal test set shows that prediction\-informed SJF can reduce average waiting time relative to FCFS, but full backfilling simulation remains future work\. The main contribution is therefore a rigorous benchmark and diagnostic analysis showing why leakage\-safe, time\-aware, and scheduler\-aware evaluation is necessary for grid runtime prediction\.

## References

- \[1\]A\. Silberschatz, P\. B\. Galvin, and G\. Gagne\(2013\)Operating system concepts\.8 edition,John Wiley & Sons\.Cited by:[§I](https://arxiv.org/html/2609.13701#S1.p1.1),[§II\-A](https://arxiv.org/html/2609.13701#S2.SS1.p3.1)\.
- \[2\]H\. Amur, G\. Shenoy, D\. Sarma, and S\. Vaddagiri\(2008\)Plimsoll: a dvs algorithm hierarchy\.pp\.\.Cited by:[§I](https://arxiv.org/html/2609.13701#S1.p2.1)\.
- \[3\]A\. Negi and P\. K\. Kumar\(2005\)Applying machine learning techniques to improve linux process scheduling\.InTENCON 2005 \- 2005 IEEE Region 10 Conference,Vol\.,pp\. 1–6\.External Links:[Document](https://dx.doi.org/10.1109/TENCON.2005.300837)Cited by:[§I](https://arxiv.org/html/2609.13701#S1.p5.1),[§II\-B](https://arxiv.org/html/2609.13701#S2.SS2.p1.1)\.
- \[4\]D\. A\. Shulga, A\. A\. Kapustin, A\. A\. Kozlov, A\. A\. Kozyrev, and M\. M\. Rovnyagin\(2016\)The scheduling based on machine learning for heterogeneous cpu/gpu systems\.In2016 IEEE NW Russia Young Researchers in Electrical and Electronic Engineering Conference \(EIConRusNW\),Vol\.,pp\. 345–348\.External Links:[Document](https://dx.doi.org/10.1109/EIConRusNW.2016.7448189)Cited by:[§I](https://arxiv.org/html/2609.13701#S1.p5.1),[§II\-B](https://arxiv.org/html/2609.13701#S2.SS2.p1.1)\.
- \[5\]W\. Smith, I\. Foster, and V\. Taylor\(2004\)Predicting application run times with historical information\.Journal of Parallel and Distributed Computing64\(9\),pp\. 1007–1016\.External Links:ISSN 0743\-7315,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.jpdc.2004.06.008),[Link](https://www.sciencedirect.com/science/article/pii/S0743731504000991)Cited by:[§I](https://arxiv.org/html/2609.13701#S1.p5.1),[§II\-B](https://arxiv.org/html/2609.13701#S2.SS2.p1.1)\.
- \[6\]T\. Helmy, S\. Al\-Azani, and O\. Bin\-Obaidellah\(2015\)A machine learning\-based approach to estimate the cpu\-burst time for processes in the computational grids\.In2015 3rd International Conference on Artificial Intelligence, Modelling and Simulation \(AIMS\),Vol\.,pp\. 3–8\.External Links:[Document](https://dx.doi.org/10.1109/AIMS.2015.11)Cited by:[§I](https://arxiv.org/html/2609.13701#S1.p5.1),[§II\-B](https://arxiv.org/html/2609.13701#S2.SS2.p1.1)\.
- \[7\]The grid workloads archive\.External Links:[Link](http://gwa.ewi.tudelft.nl/datasets/gwa-t-4-auvergrid)Cited by:[§I](https://arxiv.org/html/2609.13701#S1.p6.1),[§III\-A](https://arxiv.org/html/2609.13701#S3.SS1.p1.1)\.
- \[8\]Vandana\(2018\)Revised formula for estimating cpu burst\.Journal of International Academy of Physical Sciences22\(4\),pp\. 345–354\.Cited by:[§II\-A](https://arxiv.org/html/2609.13701#S2.SS1.p3.1)\.
- \[9\]M\. R\. Mahesh Kumar, B\. R\. Rajendra, C\. K\. Niranjan, and M\. Sreenatha\(2014\)Prediction of length of the next cpu burst in sjf scheduling algorithm using dual simplex method\.InSecond International Conference on Current Trends In Engineering and Technology \- ICCTET 2014,Vol\.,pp\. 248–252\.External Links:[Document](https://dx.doi.org/10.1109/ICCTET.2014.6966296)Cited by:[§II\-A](https://arxiv.org/html/2609.13701#S2.SS1.p4.1)\.
- \[10\]A\. Pourali and A\. M\. Rahmani\(2009\)A fuzzy\-based scheduling algorithm for prediction of next cpu\-burst time to implement shortest process next\.In2009 International Association of Computer Science and Information Technology \- Spring Conference,Vol\.,pp\. 217–220\.External Links:[Document](https://dx.doi.org/10.1109/IACSIT-SC.2009.83)Cited by:[§II\-A](https://arxiv.org/html/2609.13701#S2.SS1.p4.1)\.
- \[11\]D\. Loreti, D\. Leone, and A\. Borghesi\(2026\)Duration\-informed workload scheduler\.External Links:2604\.09599,[Link](https://arxiv.org/abs/2604.09599)Cited by:[§II\-C](https://arxiv.org/html/2609.13701#S2.SS3.p1.1)\.
- \[12\]E\. Kolker\-Hicks, D\. Zhang, and D\. Dai\(2023\)A reinforcement learning based backfilling strategy for hpc batch jobs\.InProceedings of the SC ’23 Workshops of the International Conference on High Performance Computing, Network, Storage, and Analysis,SC\-W ’23,New York, NY, USA,pp\. 1316–1323\.External Links:ISBN 9798400707858,[Link](https://doi.org/10.1145/3624062.3624201),[Document](https://dx.doi.org/10.1145/3624062.3624201)Cited by:[§II\-C](https://arxiv.org/html/2609.13701#S2.SS3.p1.1)\.
- \[13\]K\. Menear, A\. Nag, J\. Perr\-Sauer, M\. Lunacek, K\. Potter, and D\. Duplyakin\(2023\)Mastering hpc runtime prediction: from observing patterns to a methodological approach\.InPractice and Experience in Advanced Research Computing 2023: Computing for the Common Good,PEARC ’23,New York, NY, USA,pp\. 75–85\.External Links:ISBN 9781450399852,[Link](https://doi.org/10.1145/3569951.3593598),[Document](https://dx.doi.org/10.1145/3569951.3593598)Cited by:[§II\-C](https://arxiv.org/html/2609.13701#S2.SS3.p1.1)\.
- \[14\]M\. Maiterth, W\. H\. Brewer, J\. S\. Kuruvella, A\. Dey, T\. Z\. Islam, R\. Kabir, K\. Menear, D\. Duplyakin, T\. Patki, T\. Jones, and F\. Wang\(2025\)HPC digital twins for evaluating scheduling policies, incentive structures and their impact on power and cooling\.InProceedings of the SC ’25 Workshops of the International Conference for High Performance Computing, Networking, Storage and Analysis,SC Workshops ’25,New York, NY, USA,pp\. 1959–1969\.External Links:ISBN 9798400718717,[Link](https://doi.org/10.1145/3731599.3767559),[Document](https://dx.doi.org/10.1145/3731599.3767559)Cited by:[§II\-C](https://arxiv.org/html/2609.13701#S2.SS3.p1.1)\.
- \[15\]Y\. Zhang, Y\. Kurkure, Y\. Tao, M\. E\. Papka, and Z\. Lan\(2025\)A real\-time digital twin for adaptive scheduling\.External Links:2512\.18894,[Link](https://arxiv.org/abs/2512.18894)Cited by:[§II\-C](https://arxiv.org/html/2609.13701#S2.SS3.p1.1)\.
- \[16\]K\. Cetinski and M\. B\. Juric\(2015\)AME\-wpc: advanced model for efficient workload prediction in the cloud\.Journal of Network and Computer Applications55,pp\. 191–201\.External Links:ISSN 1084\-8045,[Document](https://dx.doi.org/https%3A//doi.org/10.1016/j.jnca.2015.06.001),[Link](https://www.sciencedirect.com/science/article/pii/S1084804515001241)Cited by:[§III\-A](https://arxiv.org/html/2609.13701#S3.SS1.p1.1)\.

Similar Articles

Beyond Prediction: Tail-Aware Scheduling for LLM Inference

arXiv cs.LG

This paper introduces a distribution-aware, prediction-free scheduling framework for LLM inference that replaces explicit length prediction with soft priority boosting using statistical signals. The method co-optimizes scheduling and cache-aware preemption to reduce tail latency, achieving up to 35-50% reduction in P99 TTLT compared to SRPT with perfect length knowledge.