Discover THUDM/OAG-BERT Open Source project by @THUDM

Stars
76
Rank 420,374 (Top 9 %)
Language
License
MIT License
Created over 3 years ago
Updated over 2 years ago

THUDM

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

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

ChatGLM-6B

ChatGLM-6B: An Open Bilingual Dialogue Language Model | 开源双语对话语言模型

Python

40,459

ChatGLM2-6B

ChatGLM2-6B: An Open Bilingual Chat LLM | 开源双语对话语言模型

Python

15,702

ChatGLM3

ChatGLM3 series: Open Bilingual Chat LLMs | 开源双语对话语言模型

Python

13,366

CodeGeeX

CodeGeeX: An Open Multilingual Code Generation Model (KDD 2023)

Python

8,150

CogVideo

text and image to video generation: CogVideoX (2024) and CogVideo (ICLR 2023)

Python

7,976

GLM-130B

GLM-130B: An Open Bilingual Pre-Trained Model (ICLR 2023)

Python

7,653

CodeGeeX2

CodeGeeX2: A More Powerful Multilingual Code Generation Model

Python

7,622

CogVLM

a state-of-the-art-level open visual language model | 多模态预训练模型

Python

5,913

GLM-4

GLM-4 series: Open Multilingual Multimodal Chat LMs | 开源多语言多模态对话模型

Python

4,826

VisualGLM-6B

Chinese and English multimodal conversational language model | 多模态中英双语对话语言模型

Python

4,076

GLM

GLM (General Language Model)

Python

3,168

AgentBench

A Comprehensive Benchmark to Evaluate LLMs as Agents (ICLR'24)

Python

2,144

CogVLM2

GPT4V-level open-source multi-modal model based on Llama3-8B

Python

2,018

P-tuning-v2

An optimized deep prompt tuning strategy comparable to fine-tuning across scales and tasks

Python

1,968

CogDL

CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)

Python

1,720

CogView

Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".

Python

1,691

WebGLM

WebGLM: An Efficient Web-enhanced Question Answering System (KDD 2023)

Python

1,557

AgentTuning

AgentTuning: Enabling Generalized Agent Abilities for LLMs

Python

1,339

CodeGeeX4

CodeGeeX4-ALL-9B, a versatile model for all AI software development scenarios, including code completion, code interpreter, web search, function calling, repository-level Q&A and much more.

Python

1,271

ImageReward

[NeurIPS 2023] ImageReward: Learning and Evaluating Human Preferences for Text-to-image Generation

Python

1,117

LongWriter

LongWriter: Unleashing 10,000+ Word Generation from Long Context LLMs

Python

1,076

SwissArmyTransformer

SwissArmyTransformer is a flexible and powerful library to develop your own Transformer variants.

Python

966

CogView2

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

Python

944

P-tuning

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

Python

915

LongBench

[ACL 2024] LongBench: A Bilingual, Multitask Benchmark for Long Context Understanding

Python

629

AutoWebGLM

An LLM-based Web Navigating Agent (KDD'24)

Python

584

GATNE

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

Python

522

GraphMAE

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

Python

462

CogQA

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

Python

456

Inf-DiT

Official implementation of Inf-DiT: Upsampling Any-Resolution Image with Memory-Efficient Diffusion Transformer

Python

366

GCC

GCC: Graph Contrastive Coding for Graph Neural Network Pre-Training @ KDD 2020

Python

322

MathGLM

Official Pytorch Implementation for MathGLM

Python

316

HGB

Revisiting, benchmarking, and refining Heterogeneous Graph Neural Networks.

Python

301

AlignBench

大模型多维度中文对齐评测基准 (ACL 2024)

Python

295

ComiRec

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

Python

278

LongCite

LongCite: Enabling LLMs to Generate Fine-grained Citations in Long-context QA

Python

272

RelayDiffusion

The official implementation of "Relay Diffusion: Unifying diffusion process across resolutions for image synthesis" [ICLR 2024 Spotlight]

Python

262

KOBE

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

Python

237

NLP4Rec-Papers

Paper list of NLP for recommender systems

225

ProNE

Source code and dataset for IJCAI 2019 paper "ProNE: Fast and Scalable Network Representation Learning"

Python

225

Chinese-Transformer-XL

Python

218

GRAND

Source code and dataset of the NeurIPS 2020 paper "Graph Random Neural Network for Semi-Supervised Learning on Graphs"

Python

203

LongAlign

[EMNLP 2024] LongAlign: A Recipe for Long Context Alignment of LLMs

Python

199

icetk

A unified tokenization tool for Images, Chinese and English.

CogCoM

ReST-MCTS

ReST-MCTS*: LLM Self-Training via Process Reward Guided Tree Search (NeurIPS 2024)

Python

146

KBRD

Towards Knowledge-Based Recommender Dialog System @ EMNLP 2019

Python

134

GraphMAE2

GraphMAE2: A Decoding-Enhanced Masked Self-Supervised Graph Learner in WWW'23

Python

133

iPrompt

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

Python

121

ProteinLM

Protein Language Model

Python

111

MCNS

Source code and dataset for KDD 2020 paper "Understanding Negative Sampling in Graph Representation Learning"

Python

111

VisualAgentBench

Towards Large Multimodal Models as Visual Foundation Agents

Python

CogView3

text to image to generation: CogView3-Plus and CogView3(ECCV 2024)

Python

grb

Graph Robustness Benchmark: A scalable, unified, modular, and reproducible benchmark for evaluating the adversarial robustness of Graph Machine Learning.

Python

GraphSGAN

Implementation of "GraphSGAN", a GAN-based semi-supervised learning algorithm for graph data.

Python

kgTransformer

kgTransformer: pre-training for reasoning over complex KG queries (KDD 22)

Python

ScenarioMeta

Source code and dataset for KDD 2019 paper "Sequential Scenario-Specific Meta Learner for Online Recommendation"

Python

ChatGLM-Math

Python

CogKR

Source code and dataset for paper "Cognitive Knowledge Graph Reasoning for One-shot Relational Learning"

Python

SelfKG

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

Python

FewNLU

Python

SciGLM

SciGLM: Training Scientific Language Models with Self-Reflective Instruction Annotation and Tuning (NeurIPS D&B Track 2024)

Python

Multilingual-GLM

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

Python

XDAI

Python

CogAgent

OAG

Source code and dataset for KDD 2019 paper "OAG: Toward Linking Large-scale Heterogeneous Entity Graphs"

Python

NaturalCodeBench

Python

LVBench

LVBench: An Extreme Long Video Understanding Benchmark

Python

AutoRE

Python

Graph-Reading-Group

Daily reading group on graphs at KEG

SCR

SCR: Training Graph Neural Networks with Consistency Regularization

Python

WhoIsWho

KDD'23 Web-Scale Academic Name Disambiguation: the WhoIsWho Benchmark, Leaderboard, and Toolkit

Python

FastLDM

Inference speed-up for stable-diffusion (ldm) with TensorRT.

Python

GraphCAD

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

Python

GRAND-plus

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

Python

KDD-Industrial-Papers

A list of recent industrial papers in KDD'16–'18

ApeGNN

ApeGNN: Node-Wise Adaptive Aggregation in GNNs for Recommendation (WWW'23)

Python

GLM-iprompt

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

Python

GIAAD

Graph Injection Adversarial Attack & Defense Dataset , extracted from KDD CUP 2020 ML2 Track

Python

Tsinghua-ML-Course

Course Materials for ML Course at Tsinghua

HTML

HOSMEL

A task relevant entity linking toolkit

Python

Self-Contrast

Extensive Self-Contrast Enables Feedback-Free Language Model Alignment

Python

RecDCL

RecDCL: Dual Contrastive Learning for Recommendation (WWW'24, Oral)

Python

tdgia

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

Python

BatchSampler

The source code for BatchSampler that accepted in KDD'23

Python

MRT

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

LargeScale

Python

eTrust

Source code and dataset for TKDE 2019 paper “Trust Relationship Prediction in Alibaba E-Commerce Platform”

C++

MSAGPT

Python

whoiswho-top-solutions

Python

paper-source-trace

Python

Efficient-Head-Finetuning

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

Python

IGB

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

Python

BattleAgentBench

Python

GraphAlign

GraphAlign: Pretraining One Graph Neural Network on Multiple Graphs via Feature Alignment

Python

APAR

APAR: LLMs Can Do Auto-Parallel Auto-Regressive Decoding

Python

scholar-profiling

Jupyter Notebook

citation-prediction

Python

OpenWebAgent

A convenient framework for developing LLM- and LMM-based web agents.

JavaScript

100

OAG-AQA

Python

THUDM/OAG-BERT

THUDM

Reviews

Repository Details

More Repositories