Search papers, labs, and topics across Lattice.
This paper introduces Data-Informed Centroid Splitting (DICS), a novel framework that enhances the efficiency of decision tree training by utilizing clustering techniques to create a compact set of candidate splits. By leveraging data-driven priors, DICS significantly narrows the split search space without compromising the predictive accuracy of classification trees. Extensive empirical evaluations confirm that DICS reduces training time while maintaining comparable performance across various datasets, making it a valuable advancement for scalable decision tree learning.
Training decision trees just got faster鈥擠ICS cuts down on search time without sacrificing accuracy by using data-informed clustering to streamline split selection.
Decision tree-based models are widely used in machine learning due to their interpretability and strong empirical performance. However, training decision trees can be computationally expensive, particularly for large and high-dimensional datasets, largely due to the exhaustive search over candidate splits at each node. To improve computational efficiency, we propose Data-Informed Centroid Splitting (DICS), a clustering-based framework that constructs a compact and informative set of candidate splits using data-driven priors. By incorporating class-aware structure, DICS significantly reduces the split search space for classification tasks while preserving predictive performance. We further provide theoretical analysis showing that under the stated assumptions, DICS does not degrade the performance of classification trees compared to exhaustive split search. DICS can be incorporated into classification trees, random forests, and gradient-boosting models. Extensive experiments demonstrate that DICS achieves comparable accuracy while substantially reducing training time across synthetic and benchmark datasets, highlighting the benefit of integrating data-informed priors into split selection for scalable classification tree learning.
