PiNUI: A Dataset of Protein-Protein Interactions for Machine Learning

Geoffroy Dubourg-Felonneau, Eyal Akiva, Daniel Wesego, Ranjani Varadan

Keywords: PPI, Neural Networks, Dataset, Negative interactions

TL;DR: A novel dataset to improve PPI predictions

Abstract: We introduce a new novel dataset named PiNUI: Protein Interactions with Nearly Uniform Imbalance. PiNUI is a dataset of Protein-Protein Interactions (PPI) specifically designed for Machine Learning (ML) applications that offer a higher degree of representativeness of real-world PPI tasks compared to existing ML-ready PPI datasets. We achieve such by increasing the data size and quality, and minimizing the sampling bias of negative interactions. We demonstrate that models trained on PiNUI almost always outperform those trained on conventional PPI datasets when evaluated on various general PPI tasks using external test sets.

Submission Number: 32