PiShield: A PyTorch Package for Learning with Requirements

PiShield: A PyTorch Package for Learning with Requirements

Mihaela C. Stoian, Alex Tatomir, Thomas Lukasiewicz, Eleonora Giunchiglia

Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence
Demo Track. Pages 8805-8809. https://doi.org/10.24963/ijcai.2024/1037

Deep learning models have shown their strengths in various application domains, however, they often struggle to meet safety requirements for their outputs. In this paper, we introduce PiShield, the first package ever allowing for the integration of the requirements into the neural networks' topology. PiShield guarantees compliance with these requirements, regardless of input. Additionally, it allows for integrating requirements both at inference and/or training time, depending on the practitioners' needs. Given the widespread application of deep learning, there is a growing need for frameworks allowing for the integration of the requirements across various domains. Here, we explore three application scenarios: functional genomics, autonomous driving, and tabular data generation.
Keywords:
Machine Learning: ML: Neuro-symbolic methods
Machine Learning: ML: Knowledge-aided learning
Machine Learning: ML: Structured prediction
AI Ethics, Trust, Fairness: ETF: Safety and robustness