CuBAS Improves Data Sampling for Supervised Classification

Alexandre L. M. Levada· July 7, 2026 View original

Summary

This paper introduces CuBAS (Curvature-Based Adaptive Sampling), an information-geometric framework for adaptive data selection in supervised classification. CuBAS uses local curvature, derived from a q-state Potts Markov random field model, to identify maximally informative training subsets, consistently outperforming random and uncertainty-based sampling across numerous benchmarks.

The quality of a training dataset is as crucial as its size for effective machine learning, yet many sampling methods overlook the inherent geometric structure of the data distribution. This research presents CuBAS (Curvature-Based Adaptive Sampling), a novel information-geometric framework for intelligently selecting data in supervised classification tasks. CuBAS operates on the principle that a labeled dataset can be viewed as a statistical manifold. Within this manifold, local curvature, estimated using the ratio of second to first-order observed Fisher information, accurately reflects the geometric complexity of the data. By constructing a k-nearest-neighbor graph, a closed-form curvature score is derived for each data point. This curvature signal allows CuBAS to partition the data into low-curvature regions (smooth, homogeneous clusters) and high-curvature regions (decision boundaries that are highly informative for classification). By strategically selecting data points from both types of regions, CuBAS builds compact yet maximally informative training subsets. Empirical evaluations across over 60 benchmark datasets demonstrate that CuBAS consistently and significantly outperforms random sampling and uncertainty-based baselines, regardless of labeling budget or classifier architecture. It is also computationally efficient, scaling linearly with the number of k-NN graph edges.

Why it matters

CuBAS enables the creation of more compact and informative training datasets, significantly reducing training time and computational resources while often improving model accuracy, especially in scenarios with limited labeling budgets.

How to implement this in your domain

  1. 1Integrate CuBAS into your data preprocessing pipeline for supervised classification tasks, particularly with large datasets.
  2. 2Evaluate CuBAS against existing sampling strategies to optimize training efficiency and model performance.
  3. 3Apply CuBAS in active learning scenarios to intelligently select data for human annotation, maximizing the value of labeling efforts.
  4. 4Explore the use of information-geometric concepts like curvature for other data analysis and machine learning tasks.

Who benefits

HealthcareFinanceE-commerceMarketingAutonomous Vehicles

Key takeaways

  • Data informativeness is crucial, not just size, for training ML models.
  • CuBAS uses information-geometric curvature to identify informative data points.
  • It selects data from both homogeneous clusters and decision boundaries.
  • CuBAS consistently outperforms other sampling methods, improving efficiency and accuracy.

Original post by Alexandre L. M. Levada

"arXiv:2607.03145v1 Announce Type: new Abstract: The informativeness of a training set is as consequential as its size, yet most sampling strategies remain agnostic to the intrinsic geometry of the data distribution. We introduce CuBAS (Curvature-Based Adaptive Sampling), an infor…"

View on X

Originally posted by Alexandre L. M. Levada on X · view source

Want to go deeper?

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

Explore courses