Search papers, labs, and topics across Lattice.
This paper introduces a feature-scoped machine learning approach to predict slot-time consumption for queries in cloud data warehouses, addressing the challenges of variable query costs in multi-tenant environments. By leveraging pre-execution observable signals such as structured query complexity scores, data volume features, and textual features, the model effectively estimates costs without relying on runtime factors. The approach achieves a significant reduction in mean absolute error for cost-significant queries, demonstrating its potential to improve budget management and scheduling efficiency in cloud environments.
Predicting query slot-time consumption with a machine learning model reduces cost estimation errors by up to 37%, revolutionizing budget management in cloud data warehouses.
Cloud data warehouses bill compute based on slot-time consumed. In shared multi-tenant environments, query cost is highly variable and hard to estimate before execution, causing budget overruns and degraded scheduling. Static query-planner heuristics fail to capture complex SQL structure, data skew, and workload contention. We present a feature-scoped machine learning approach that predicts BigQuery slot-time before execution using only pre-execution observable signals: a structured query complexity score derived from SQL operator costs, data volume features from planner estimates and workload metadata, and textual features from query text. We deliberately exclude runtime factors (slot-pool utilization, cache state, realized skew) unknowable at submission. The model uses a HistGradientBoostingRegressor trained on log-transformed slot-time, with a TF-IDF + TruncatedSVD-512 text pipeline fused with numeric and categorical features. Trained on 749 queries across seven deployment environments and evaluated out-of-distribution on 746 queries from two held-out environments, the model achieves MAE 1.17 slot-minutes, RMSE 4.71, and 74% explained variance on the full workload. On cost-significant queries (slot-time>= 0.01 min, N=282) the model achieves MAE 3.10 versus 4.95 for a predict-mean baseline and 4.54 for predict-median, a 30-37% reduction. On long-tail queries (>= 20 min, N=22) the model does not outperform trivial baselines, consistent with the hypothesis that long-tail queries are dominated by unobserved runtime factors outside the current feature scope. A complexity-routed dual-model architecture is described as a practical refinement, and directions for closing the long-tail gap are identified as future work.