csirmaz / trained-linearization

Interpreting neural networks by reducing nonlinearities during training

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Interpreting Neural Networks by Reducing Nonlinearities during Training

This repo contains a short paper and sample code demonstrating a simple solution that makes it possible to extract rules from a neural network that employs Parametric Rectified Linear Units (PReLUs). We introduce a force, applied in parallel to backpropagation, that aims to reduce PReLUs into the identity function, which then causes the neural network to collapse into a smaller system of linear functions and inequalities suitable for review or use by human decision makers.

As this force reduces the capacity of neural networks, it is expected to help avoid overfitting as well.

Download the article in PDF format from the latest release at https://github.com/csirmaz/trained-linearization/releases/latest .

About

Interpreting neural networks by reducing nonlinearities during training

License:MIT License


Languages

Language:TeX 45.8%Language:Lua 33.8%Language:HTML 17.0%Language:Perl 3.4%