Learning Apparatus Mitigating Membership Inference Attacks via Training Data Modification

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Machine learning models in the process of learning are vulnerable to membership inference attacks, leading to a risk of information leakage due to overfitting, as they mistakenly learn in the direction of the target variable during distillation using output data from pre-trained models.

Innovation Solution

A learning apparatus and method that generates first training data using unlabeled data as input to a pre-learned model, and second training data with modified target variables to enhance the magnitude difference between elements, thereby generating a machine learning model tolerant to membership inference attacks during learning.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Reliability

If distillation is used to compress the machine learning model and prevent overfitting, then the final model becomes tolerant to membership inference attacks, but the model during the learning process becomes overfitted and vulnerable to membership inference attacks

Engineering Contradiction:
Improvetolerance to membership inference attacksVSAvoidinformation leakage during learning process
Core Design Contradiction:
ReliabilityVSObject-affected harmful factors

Solution Approach 1:

The patent applies preliminary action by modifying the training data before the distillation learning process begins. Specifically, it generates artificial training data with controlled target variable distributions that prevent overfitting from occurring during the learning process. This preliminary data preparation ensures that even during intermediate learning stages, the model remains resistant to membership inference attacks, rather than only achieving this property after complete training.

Inventive Principle:
Principle #10Preliminary action

Solution Approach 2:

The patent implements preliminary anti-action by introducing a countermeasure against membership inference attacks before the harmful effect can manifest. By modifying the target variable distribution in advance during data generation, the system prevents the model from developing vulnerabilities to membership inference attacks during the learning process, rather than attempting to fix the problem after overfitting occurs.

Inventive Principle:
Principle #9Preliminary anti-action

2Productivity

If the output of a pre-learned model is used as a target variable for distillation, then model compression is achieved, but the learning process mistakenly proceeds in the direction of the target variable in the original training data

Engineering Contradiction:
Improvemodel compression efficiencyVSAvoidlearning direction accuracy
Core Design Contradiction:
ProductivityVSManufacturing precision

Solution Approach 1:

The patent applies parameter changes by modifying the distribution parameters of the target variable during training data generation. Instead of using the original target variable distribution from the pre-learned model, it generates a new distribution with adjusted parameters that prevent the model from mistakenly learning in the direction of the original training data targets. This allows distillation to proceed effectively while maintaining correct learning direction.

Inventive Principle:
Principle #35Parameter changes

Solution Approach 2:

The patent introduces an intermediary element - artificially generated training data with controlled target variable distributions - that mediates between the pre-learned model's output and the final distilled model. This intermediary data layer allows the distillation process to compress the model while preventing direct overfitting to the original training data targets, as the intermediary data provides a controlled learning signal.

Inventive Principle:
Principle #24Intermediary (Mediator)

Data Source

PatentUS20240211808A1Learning apparatus, learning method, and program
Publication Date: 2024.06.27 NEC CORP
  • US20240211808A1 patent drawing
  • US20240211808A1 patent drawing
  • US20240211808A1 patent drawing

AI summary

A learning apparatus according to the present disclosure includes: a first training data generating unit configured to generate first training data in which a vector including values of a plurality of elements output by inputting unlabeled training data to a pre-learned machine learning model is a target variable; a second training data generating unit configured to generate second training data in which the values of the elements of the vector that is the target variable of the first training data are set so that a difference in magnitude of value between at least some of the elements becomes larger; and a learning unit configured to generate a machine learning model by machine learning using the first training data and the second training data.