Skip to content
This repository was archived by the owner on Sep 5, 2025. It is now read-only.

Comments

Add proof of concept for model inversion attack#71

Merged
suhacker1 merged 84 commits intotrailofbits:masterfrom
pwang00:master
Nov 26, 2021
Merged

Add proof of concept for model inversion attack#71
suhacker1 merged 84 commits intotrailofbits:masterfrom
pwang00:master

Conversation

@pwang00
Copy link
Contributor

@pwang00 pwang00 commented Jun 13, 2021

This proof of concept represents a minimally working model inversion attack on a classifier trained on the MNIST dataset. Work needs to be done to convert this PoC into a generalize API.

@suhacker1 suhacker1 merged commit c5b8648 into trailofbits:master Nov 26, 2021
diegom4riano pushed a commit to diegom4riano/PrivacyRaven that referenced this pull request May 20, 2025
* Add barebones examples

* Update examples

* Keep fork up to date

* Bump PL version for more metric functions

* Create example with new model and enable custom Trainer args

* Comment more

* Replace examples with new classifier

* Update documentation

* Update tests with new MNIST model

* Replace classifier in synthesis tests

* Remove redundant space

* Resolve linting issues

* Attempt to fix copycat test errors

* Remove test deadlines and failing assertion

* Resolve linting errors

* Update examples

* Keep fork up to date

* Update examples

* Update examples

* Final examples

* Remove create_model.py

* Remove create_model.py from README.md

* Update all examples

* Test support for python3.6

* Updated PrivacyRaven tests and added docker support

* Update Dockerfile to reflect directory changes

* Update directory structure

* Update directory structure

* Remove backups

* Update noxfile

* Remove noxfile backup

* Update README.md

* Add Dockerfile and build script

* Update examples and requirements

* Update examples to resolve GPU issues on colab

* Update comments

* Update example code

* Change default arg for gpus

* Update formatting to pass linting

* Update README.md

* Update PoC for Model Inversion

* Update inversion files

* Update core inversion

* Update rest of files

* Fix typo

* Update core files

* Update inversion

* Add inversion model

* Update core inversion

* Update inversion files

* Update inversion

* Update core inversion

* Further update inversion

* Update inversion

* Update inversion

* Add inversion model

* Update inversion files

* Temp fix cuda errors

* add current inversion changes

* Finally fix GPU bug

* Allow inversion model to be customized

* Fix syntax error

* Remove unnecessary params

* Remove duplicate argument

* Update inversion files, add better plotting

* Improve proof of concept

* Improve proof of concept

* Further update PoC

* Add more inversion results

* Fix failing examples

* Clean up debug info

* Fix merge conflicts, add softmax to four_layer_classifier.py to account for input validation

* Fix merge conflicts

Co-authored-by: Suha S. Hussain <suhashussain1@gmail.com>
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants