Learning Apparatus Mitigating Membership Inference Attacks via Training Data Modification
Find Innovative SolutionsGenerate 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
Engineering 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
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.
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.
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
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.
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.
Data Source
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.


