Homogeneous Train-Test Splitting Enhances AutoML Evaluation

Yearn Tan Yin Tze, Charles Grellois· July 30, 2026 View original

Summary

This paper investigates the impact of statistical similarity in train-test splitting on AutoML model evaluation, proposing an "Optimised-Distribution" method. It finds that geometry-based splitting methods can introduce instability, while the new method achieves the highest mean Maximum Mean Discrepancy (MMD) similarity, improving evaluation accuracy.

This research highlights the critical role of train-test splitting methods in achieving accurate model evaluation within Automated Machine Learning (AutoML). Standard random splitting often assumes identical underlying distributions between partitions, an assumption frequently violated in real-world datasets characterized by class imbalance, clustering, or spatial autocorrelation. The study compares five established splitting strategies, including random, stratified, Kennard-Stone, Duplex, and SPXY, across fifteen UCI benchmark datasets. The analysis reveals that geometry-based methods, while aiming for coverage, can inadvertently introduce instability into downstream performance estimates by yielding near-zero Maximum Mean Discrepancy (MMD) scores, indicating a lack of diversity. To counter this, the paper proposes an "Optimised-Distribution" method that explicitly treats statistical similarity as an optimization objective. This new method consistently achieves the highest mean MMD similarity (89.0%) among all evaluated strategies, suggesting a more robust approach to creating homogeneous train-test splits for reliable AutoML model evaluation.

Why it matters

Data scientists and ML engineers can significantly improve the reliability and accuracy of their model evaluations, especially in AutoML pipelines, by adopting more statistically homogeneous train-test splitting methods.

How to implement this in your domain

  1. 1Review current train-test splitting practices in AutoML pipelines to ensure statistical similarity between partitions.
  2. 2Implement the proposed "Optimised-Distribution" method or similar strategies to create more homogeneous splits.
  3. 3Utilize statistical tests like MMD to quantify and monitor the similarity between training and testing datasets.
  4. 4Educate data science teams on the potential pitfalls of standard random splitting for datasets with inherent biases or structures.

Who benefits

AI DevelopmentData ScienceSoftware TestingQuality Assurance

Key takeaways

  • Accurate model evaluation depends on homogeneous train-test splitting, often violated by random methods.
  • Geometry-based splitting can introduce instability in performance estimates.
  • The "Optimised-Distribution" method explicitly optimizes for statistical similarity, achieving high MMD scores.
  • Adopting better splitting methods enhances AutoML model evaluation reliability.

Original post by Yearn Tan Yin Tze, Charles Grellois

"arXiv:2607.26625v1 Announce Type: new Abstract: Accurate model evaluation in machine learning depends critically on how datasets are split into training and testing subsets. Standard random splitting assumes that both partitions share the same underlying distribution, an assumpti…"

View on X

Originally posted by Yearn Tan Yin Tze, Charles Grellois on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses