Decision Tree Dimension Reduction Optimization

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Existing decision tree models, such as Classification and Regression Trees (CART), rely on greedy criteria or heuristics for splitting, which can lead to suboptimal performance due to computational constraints, especially in high-dimensional data settings.

Innovation Solution

An end-to-end machine learning framework that integrates dimension reduction techniques, like principal component analysis (PCA), into the decision tree training process, optimizing routing functions and splits simultaneously across all nodes using optimization algorithms, and employing regularizers to enhance performance.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Productivity

If greedy criterion or heuristics are used for splitting in decision trees, then computational cost is reduced, but splitting optimality is sacrificed

Engineering Contradiction:
Improvecomputational efficiencyVSAvoidsplitting optimality
Core Design Contradiction:
ProductivityVSReliability

Solution Approach 1:

The patent merges dimension reduction and splitting optimization into a unified framework where both operations are performed simultaneously rather than sequentially. This integration allows the model to achieve optimal splitting while managing computational complexity through coordinated optimization of multiple components.

Inventive Principle:
Principle #5Merging (Combining)

Solution Approach 2:

The patent transforms the high-dimensional feature space into a lower-dimensional representation by learning optimal projection parameters. This parameter transformation enables the model to work with reduced dimensions while maintaining or improving splitting quality through learned feature transformations.

Inventive Principle:
Principle #35Parameter changes

2Reliability

If dimension reduction is performed on high-dimensional data, then model performance and scalability are improved, but computational complexity of the training process increases

Engineering Contradiction:
Improveout-of-sample performanceVSAvoidtraining computational complexity
Core Design Contradiction:
ReliabilityVSDevice complexity

Solution Approach 1:

The patent segments the training process into distinct phases: dimension reduction phase and splitting optimization phase. This segmentation allows each component to be optimized independently and trained iteratively, reducing the overall computational burden compared to simultaneous optimization of all parameters.

Inventive Principle:
Principle #1Segmentation

Solution Approach 2:

The patent employs dynamic optimization where the dimension reduction parameters and splitting parameters are updated iteratively in an alternating fashion. This dynamic approach allows the model to adaptively adjust parameters during training, achieving good performance while managing computational complexity through staged optimization.

Inventive Principle:
Principle #15Dynamics

Data Source

PatentUS20230186107A1Boosting classification and regression tree performance with dimension reduction
Publication Date: 2023.06.15 INTERNATIONAL BUSINESS MACHINE CORPORATION
  • US20230186107A1 patent drawing
  • US20230186107A1 patent drawing
  • US20230186107A1 patent drawing

AI summary

A system and method can be provided for constructing and training a decision tree for machine learning. A training set can be received. The decision tree can be initialized by constructing a root node and a root solver can be trained with the training set. A processor can grow the decision tree by iteratively splitting nodes of the decision tree, where at a node of the decision tree, dimension reduction is performed on features of data of the training set received at the node, and the data having reduced dimension is split based on a routing function, for routing to another node of the decision tree. The dimension reduction and the split can be performed together at the node based on solving a nonlinear optimization problem.