Cluster-Based Data Augmentation for Imbalanced Machine Learning Datasets

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

In machine learning, especially with small or imbalanced datasets, data augmentation is necessary to improve prediction accuracy, but existing methods can be time-consuming and may not effectively enhance the quality and variety of data, potentially leading to overfitting and resource inefficiency.

Innovation Solution

A computer-implemented method for data augmentation that clusters input data into groups based on similarity, determines clusters that require augmentation, and applies specific augmentation methods to improve prediction accuracy, thereby optimizing the augmentation process and reducing computational resources.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Quantity of substance

If data augmentation is performed on the entire dataset, then the size and quality of the dataset is improved, but the time consumption and computational resources increase significantly

Engineering Contradiction:
Improvedataset sizeVSAvoidtime consumption
Core Design Contradiction:
Quantity of substanceVSLoss of time

Solution Approach 1:

The patent divides the dataset into multiple clusters based on feature similarity. Instead of applying data augmentation uniformly to the entire dataset, the method selectively applies augmentation only to specific clusters that meet certain criteria (e.g., small cluster size, high variance, or low representation). This segmentation approach reduces the overall computational burden while maintaining the benefits of data augmentation where most needed.

Inventive Principle:
Principle #1Segmentation

Solution Approach 2:

The patent applies different data augmentation strategies to different clusters based on their specific characteristics. Each cluster is evaluated independently, and augmentation is applied locally only where necessary. This local quality approach ensures that computational resources are focused on the most beneficial areas of the dataset rather than uniformly processing all data points.

Inventive Principle:
Principle #3Local quality

2Quantity of substance

If data augmentation is performed on the entire dataset, then the quality and variety of data is improved, but the computational resources and efficiency deteriorate

Engineering Contradiction:
Improvedata varietyVSAvoidcomputational efficiency
Core Design Contradiction:
Quantity of substanceVSProductivity

Solution Approach 1:

By segmenting the dataset into clusters and selectively applying augmentation only to certain clusters, the method maintains data variety in the most critical areas while preserving computational efficiency overall. The segmentation allows the system to focus resources on generating diverse data where it matters most.

Inventive Principle:
Principle #1Segmentation

Solution Approach 2:

The patent applies data augmentation partially rather than excessively to the entire dataset. By identifying and augmenting only the clusters that benefit most from this process, the method achieves sufficient data variety without the diminishing returns and resource waste that would result from applying augmentation uniformly across all data.

Inventive Principle:
Principle #16Partial or excessive action

3Stability of the object's composition

If data augmentation is applied uniformly to all clusters, then the dataset becomes more balanced, but the risk of overfitting increases due to redundant data

Engineering Contradiction:
Improvedataset balanceVSAvoidoverfitting risk
Core Design Contradiction:
Stability of the object's compositionVSReliability

Solution Approach 1:

The patent applies data augmentation locally to specific clusters rather than uniformly across all clusters. This allows the method to balance the dataset in areas where it is most needed while avoiding the creation of redundant data in clusters that already have sufficient representation. The local quality approach prevents overfitting by ensuring that augmentation is applied with precision rather than blanket coverage.

Inventive Principle:
Principle #3Local quality

4Quantity of substance

If traditional data augmentation methods are used, then the dataset size increases, but the processing time and resource usage increase proportionally

Engineering Contradiction:
Improvedataset sizeVSAvoidprocessing speed
Core Design Contradiction:
Quantity of substanceVSProductivity

Solution Approach 1:

The patent segments the dataset into clusters and processes only the relevant clusters for data augmentation. This segmentation reduces the total number of data points that require augmentation processing, thereby increasing processing speed while still achieving the desired increase in overall dataset size through targeted expansion of specific clusters.

Inventive Principle:
Principle #1Segmentation

Data Source

PatentUS11947570B2Data augmentation
Publication Date: 2024.04.02 INTERNATIONAL BUSINESS MACHINE CORPORATION
  • US11947570B2 patent drawing
  • US11947570B2 patent drawing
  • US11947570B2 patent drawing

AI summary

A computer-implemented method for data augmentation is provided according an embodiment of the present disclosure. In the method, a first feature vector for input data may be obtained based on a first model. The input data may be clustered to a plurality of clusters. For each of the clusters, a second feature vector may be obtained based on the first model. Then, a similarity between the first feature vector and the second feature vector may be estimated for each of the clusters. At least one cluster of the plurality of clusters for which the similarity is lower than a threshold may be determined. Moreover, data augmentation may be performed to the at least one cluster.