Heterogeneous Deep Graph Infomax

Last update: Oct 31, 2022

Overview

Heterogeneous-Deep-Graph-Infomax

Parameter Setting:

HDGI-A:

Node-level dimension: 16
Attention head: 4
Semantic-level attention vector: 8
learning rate: 0.02

HDGI-C:

Node-level dimension: 64
Semantic-level attention vector: 8
learning rate: 0.02

GAT:

Node-level dimension: 16
Attention head: 4
learning rate: 0.005
Drop out ratio: 0.6

GCN:

Hidden-unit dimension: 64
learning rate: 0.01
Drop out ratio: 0.5

RGCN:

Hidden-unit dimension: 16
learning rate: 0.01
Drop out ratio: 0

Metapath2Vec:

Embedding dimension: 100
learning rate: 0.01
negative-samples: 5
Window: 1

HAN:

Node-level dimension: 16
Attention head: 4
Semantic-level dimension: 8
learning rate: 0.005
Node-level Drop out ratio: 0.6
Semantic-level Drop out ratio: 0.6

Deepwalk:

Number of walks per node: 10
Dimensions of word embeddings: 128
Length of random walk: 30
Window size for skipgram: 5

DGI:

Hidden-unit dimension: 64
learning rate: 0.001
Drop out ratio: 0

Owner

GitHub Repository

[NeurIPS 2020] Official repository for the project "Listening to Sound of Silence for Speech Denoising"

Listening to Sounds of Silence for Speech Denoising Introduction This is the repository of the "Listening to Sounds of Silence for Speech Denoising" p

40 Dec 20, 2022

Experiments and code to generate the GINC small-scale in-context learning dataset from "An Explanation for In-context Learning as Implicit Bayesian Inference"

GINC small-scale in-context learning dataset GINC (Generative In-Context learning Dataset) is a small-scale synthetic dataset for studying in-context

29 Dec 19, 2022

PyTorch implementations of the paper: "DR.VIC: Decomposition and Reasoning for Video Individual Counting, CVPR, 2022"

DRNet for Video Indvidual Counting (CVPR 2022) Introduction This is the official PyTorch implementation of paper: DR.VIC: Decomposition and Reasoning

35 Nov 22, 2022

Autoregressive Models in PyTorch.

Autoregressive This repository contains all the necessary PyTorch code, tailored to my presentation, to train and generate data from WaveNet-like auto

41 Oct 09, 2022

A set of tools for converting a darknet dataset to COCO format working with YOLOX

darknet格式数据→COCO darknet训练数据目录结构（详情参见dataset/darknet）： darknet ├── class.names ├── gen_config.data ├── gen_train.txt ├── gen_valid.txt └── images

148 Jan 03, 2023

Graph Self-Attention Network for Learning Spatial-Temporal Interaction Representation in Autonomous Driving

GSAN Introduction Code for paper GSAN: Graph Self-Attention Network for Learning Spatial-Temporal Interaction Representation in Autonomous Driving, wh

6 Oct 27, 2022

TrTr: Visual Tracking with Transformer

TrTr: Visual Tracking with Transformer We propose a novel tracker network based on a powerful attention mechanism called Transformer encoder-decoder a

66 Dec 27, 2022

Skyformer: Remodel Self-Attention with Gaussian Kernel and Nystr\"om Method (NeurIPS 2021)

Skyformer This repository is the official implementation of Skyformer: Remodel Self-Attention with Gaussian Kernel and Nystr"om Method (NeurIPS 2021).

46 Sep 20, 2022

A Python library for Deep Graph Networks

PyDGN Wiki Description This is a Python library to easily experiment with Deep Graph Networks (DGNs). It provides automatic management of data splitti

194 Dec 22, 2022

Official PyTorch Implementation of "AgentFormer: Agent-Aware Transformers for Socio-Temporal Multi-Agent Forecasting".

AgentFormer This repo contains the official implementation of our paper: AgentFormer: Agent-Aware Transformers for Socio-Temporal Multi-Agent Forecast

161 Dec 23, 2022

Tiny Object Detection in Aerial Images.

AI-TOD AI-TOD is a dataset for tiny object detection in aerial images. [Paper] [Dataset] Description AI-TOD comes with 700,621 object instances for ei

116 Dec 30, 2022

The code repository for EMNLP 2021 paper "Vision Guided Generative Pre-trained Language Models for Multimodal Abstractive Summarization".

Vision Guided Generative Pre-trained Language Models for Multimodal Abstractive Summarization [Paper] accepted at the EMNLP 2021: Vision Guided Genera

42 Jan 07, 2023