AI Shanghai Lablet
Amazon Web Services
We provide Open Source Projects for Researchers and Developers
Deep Graph Learning
DGL is an easy-to-use, high performance and scalable Python package for deep learning on graphs. DGL is framework agnostic, meaning if a deep graph model is a component of an end-to-end application, the rest of the logics can be implemented in any major frameworks, such as PyTorch, Apache MXNet or TensorFlow. Learn more about DGL.
Natural Language Processing
Graphs are everywhere in text: syntactic trees, semantic graphs, discourse graphs, etc. We work on extracting the graphs from text and do reasoning over them in downstream tasks. The major challenge in graph extraction is lack of annotated data. One of our published works tackle this problem with unsupervised cycle training on non-parallel text-graph corpus, and achieves comparable results compared with supervised methods.
Image and video data contain relations among the captured objects, as well as the temporal change for each object. With graph neural networks, it’s easy to discover and represent the complex relations and changes in the data. Our first work along this line is Hilander: a GNN model for large-scale image clustering which achieves SOTA performance on multiple public datasets.