Active Learning Data Selection via Integer Programming

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Existing active learning methods for deep learning face challenges in selecting the most representative subset of data points for labeling, often resulting in sub-optimal solutions due to the use of greedy heuristics and approximations, which can lead to less accurate models and increased computational resources, especially in applications like medical imaging where data labeling is time-consuming and costly.

Innovation Solution

The use of Generalized Benders Decomposition (GBD) in integer programming to directly minimize the Wasserstein distance between the data set and the selected subset, ensuring globally optimal data selection by decomposing the optimization problem into smaller sub-problems and iteratively solving them, thereby guaranteeing convergence to a globally optimal solution within a threshold tolerance.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Device complexity

If greedy heuristics are used to approximate the optimization problem, then computational complexity is reduced and the method becomes practical for large data sets, but the solution quality deteriorates and becomes sub-optimal

Engineering Contradiction:
Improvecomputational complexityVSAvoidsolution quality
Core Design Contradiction:
Device complexityVSMeasurement precision

Solution Approach 1:

The patent segments the large-scale integer optimization problem into a master problem and multiple worker problems. The master problem determines the optimal subset size and selects candidate points, while worker problems compute local objective values and gradients. This segmentation allows the use of exact optimization methods on smaller subproblems rather than attempting to solve the entire problem at once, thus maintaining solution quality while reducing computational complexity.

Inventive Principle:
Principle #1Segmentation

Solution Approach 2:

The patent performs preliminary actions by pre-computing distance matrices between all data points and pre-selecting candidate subsets before the main optimization loop. This preliminary processing reduces the computational burden during iterative optimization and enables the use of exact methods without excessive computational cost.

Inventive Principle:
Principle #10Preliminary action

2Measurement precision

If integer programming is used to solve the optimization problem exactly, then solution quality is improved and global optimality is achieved, but computational complexity increases and the method becomes impractical for large data sets

Engineering Contradiction:
Improvesolution qualityVSAvoidcomputational complexity
Core Design Contradiction:
Measurement precisionVSDevice complexity

Solution Approach 1:

The patent divides the integer programming problem into a master problem that determines subset size and candidate selection, and worker problems that evaluate local objectives. This segmentation enables exact optimization on decomposed subproblems rather than the full problem, achieving global optimality with reduced computational burden.

Inventive Principle:
Principle #1Segmentation

Solution Approach 2:

The patent computes distance matrices and candidate evaluations for only the necessary subset of data points rather than all possible combinations. By focusing computational effort on relevant candidates and using early termination criteria, the method achieves exact optimization without exhaustive computation.

Inventive Principle:
Principle #16Partial or excessive action

3Measurement precision

If more data points are selected for labeling, then model accuracy is improved, but labeling costs and computational resources increase

Engineering Contradiction:
Improvemodel accuracyVSAvoidlabeling budget
Core Design Contradiction:
Measurement precisionVSQuantity of substance

Solution Approach 1:

The patent pre-computes distance matrices and candidate evaluations before the optimization process, enabling efficient evaluation of different labeling budgets. This preliminary action allows the system to quickly determine the optimal subset size and composition for any given budget constraint without expensive repeated computations.

Inventive Principle:
Principle #10Preliminary action

Solution Approach 2:

The patent replaces greedy mechanical selection heuristics with an exact integer programming formulation that systematically evaluates and selects the optimal subset of data points. This substitution ensures that every labeled point contributes maximally to model accuracy within the given budget, rather than relying on approximate greedy choices.

Inventive Principle:
Principle #28Mechanics substitution (Replace mechanical system)

Data Source

PatentUS20230244985A1Optimized active learning using integer programming
Publication Date: 2023.08.03 NVIDIA CORP
  • US20230244985A1 patent drawing
  • US20230244985A1 patent drawing
  • US20230244985A1 patent drawing

AI summary

In various examples, a representative subset of data points are queried or selected using integer programming to minimize the Wasserstein distance between the selected data points and the data set from which they were selected. A Generalized Benders Decomposition (GBD) may be used to decompose and iteratively solve the minimization problem, providing a globally optimal solution (an identified subset of data points that match the distribution of their data set) within a threshold tolerance. Data selection may be accelerated by applying one or more constraints while iterating, such as optimality cuts that leverage properties of the Wasserstein distance and/or pruning constraints that reduce the search space of candidate data points. In an active learning implementation, a representative subset of unlabeled data points may be selected using GBD, labeled, and used to train machine learning model(s) over one or more cycles of active learning.