THUDM

CogView2

official code repo for paper "CogView2: Faster and Better Text-to-Image Generation via Hierarchical Transformers"

P-tuning

A novel method to tune language models. Codes and datasets for paper ``GPT understands, too''.

GATNE

Source code and dataset for KDD 2019 paper "Representation Learning for Attributed Multiplex Heterogeneous Network"

CogQA

Source code and dataset for ACL 2019 paper "Cognitive Graph for Multi-Hop Reading Comprehension at Scale"

ComiRec

Source code and dataset for KDD 2020 paper "Controllable Multi-Interest Framework for Recommendation"

KOBE

Towards Knowledge-Based Personalized Product Description Generation in E-commerce @ KDD 2019

Chinese-Transformer-XL

icetk

A unified tokenization tool for Images, Chinese and English.

KBRD

Towards Knowledge-Based Recommender Dialog System @ EMNLP 2019

iPrompt

Code, Data and Demo for Paper: Controllable Generation from Pre-trained Language Models via Inverse Prompting

OAG-BERT

A heterogeneous entity-augmented academic language model based on Open Academic Graph (OAG)

FewNLU

SelfKG

Codes for WWW2022 accepted paper: SelfKG: Self-Supervised Entity Alignment in Knowledge Graphs

Multilingual-GLM

The multilingual variant of GLM, a general language model trained with autoregressive blank infilling objective

XDAI

FasterTransformer

Transformer related optimization, including BERT, GPT

SCR

SCR: Training Graph Neural Networks with Consistency Regularization

GRAND-plus

Code and dataset for paper "GRAND+: Scalable Graph Random Neural Networks"

GraphCAD

TKDE'22-GraphCAD: https://arxiv.org/pdf/2108.07516.pdf

GLM-iprompt

Apply Iprompt on GLM with innovative new methods. Currently support Chinese QA, English QA and Chinese poem generation.

HOSMEL

A task relevant entity linking toolkit

MRT

MRT: Tracing the Evolution of Scientific Publications (TKDE 2021)

tdgia

code for paper TDGIA:Effective Injection Attacks on Graph Neural Networks (KDD 2021, research track)

Efficient-Head-Finetuning

Source code for EMNLP2022 long paper: Parameter-Efficient Tuning Makes a Good Classification Head

IGB

Source code and dataset for IJCAI 2022 paper "Rethinking the Setting of Semi-supervised Learning on Graphs"

STAM

Source code and dataset for WWW'22 paper "STAM: A Spatiotemporal Aggregation Method for Graph Neural Network-based Recommendation"

DistAlign-GNNs

Paper-Rec

RecNS

Source code and dataset for TKDE'22 paper "Region or Global? A Principle for Negative Sampling in Graph-based Recommendation"

PAL

A Pre-training Framework for Adaptive Learning in MOOCs