[go: nahoru, domu]

Skip to content

hilbert-qyw/CertifiedMonotonicNetwork

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Certified Monotonic Neural Networks

This repo contains the code for NeurIPS 2020 spotlight paper:

by Xingchao Liu, Xing Han, Na Zhang, Qiang Liu from UT Austin and Tsinghua University

Requirements: python>=3.7, PyTorch>=1.0.0, Gurobi>=9.0.1

  1. Small monotonic regularization usually gets best results, but training with small monotonic regularization can be hard to guarantee monotonicity. We suggest to loop over random seeds before increasing monotonic regularization to get the best monotonic neural network.

  2. The network structure is slightly modified compared to the ones in the paper. The networks in this repo have less parameters but similar performance.

Citation:

If you use the code or our work is related to yours, please cite us:

@article{liu2020certified,
  title={Certified Monotonic Neural Networks},
  author={Liu, Xingchao and Han, Xing and Zhang, Na and Liu, Qiang},
  journal={Advances in Neural Information Processing Systems},
  volume={33},
  year={2020}
}

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%