PyKale: Knowledge-Aware Machine Learning from Multiple Sources in Python

Abstract

Machine learning is a general-purpose technology holding promises for many interdisciplinary research problems. However, significant barriers exist in crossing disciplinary boundaries when most machine learning tools are developed in different areas separately. We present Pykale - a Python library for knowledge-aware machine learning on graphs, images, texts, and videos to enable and accelerate interdisciplinary research. We formulate new green machine learning guidelines based on standard software engineering practices and propose a novel pipeline-based application programming interface (API). PyKale focuses on leveraging knowledge from multiple sources for accurate and interpretable prediction, thus supporting multimodal learning and transfer learning (particularly domain adaptation) with latest deep learning and dimensionality reduction models. We build PyKale on PyTorch and leverage the rich PyTorch ecosystem. Our pipeline-based API design enforces standardization and minimalism, embracing green machine learning concepts via reducing repetitions and redundancy, reusing existing resources, and recycling learning models across areas. We demonstrate its interdisciplinary nature via examples in bioinformatics, knowledge graph, image/video recognition, and medical imaging.

arXiv commentThis library is available at https://github.com/pykale/pykale

References in corpus (13)

• Scikit-learn: Machine Learning in Python
• Learning Transferable Features with Deep Adaptation Networks
• DeepDTA: Deep Drug-Target Binding Affinity Prediction
• Modeling polypharmacy side effects with graph convolutional networks
• Conditional Adversarial Domain Adaptation
• Fast Graph Representation Learning with PyTorch Geometric
• TorchIO: A Python library for efficient loading, preprocessing, augmentation and patch-based sampling of medical images in deep learning
• DeepAffinity: Interpretable Deep Learning of Compound-Protein Affinity through Unified Recurrent and Convolutional Neural Networks
• GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Acceleration
• Wasserstein Distance Guided Representation Learning for Domain Adaptation
• TensorLy: Tensor Learning in Python
• Deep Isometric Learning for Visual Recognition
• GripNet: Graph Information Propagation on Supergraph for Heterogeneous Graphs