Fast Learning of MNL Model From General Partial Rankings with Application to Network Formation Modeling

Last update: Aug 19, 2022

Related tags

Overview

Fast-Partial-Ranking-MNL

This repo provides a PyTorch implementation for the CopulaGNN models as described in the following paper:

Fast Learning of MNL Model From General Partial Rankings with Application to Network Formation Modeling.

Jiaqi Ma*, Xingjian Zhang*, and Qiaozhu Mei. WSDM 2022.

(*: Equal contribution.)

Requirements

The code requires the following packages.

more_itertools==8.10.0
networkx==2.5.1
numpy==1.19.5
pandas==1.1.5
pyclustering==0.10.1.2
torch==1.9.0
tqdm==4.62.3

Example Commands to Run the Experiments

Learning single MNL from partial rankings on synthetic data

python3 dag_synthetic.py --num_classes 100 --num_samples 5000  # single MNL

Learning mixture of MNL from partial rankings on synthetic data

python3 dag_synthetic.py --num_classes 60 --num_samples 5000 --alphas [1,1,1]  --init_by_cluster # 3 MNLs with clustering based init

Network formation modeling of synthetic network data

python3 network_synthetic.py -r 0.5 -p 0.5 --fof --ua --pa --loss topk  # run full model with 4 components on a mixed (r,p)-graph

Network formation modeling of Flickr & Microsoft Academic Graph

cd source
wget -4 http://socialnetworks.mpi-sws.mpg.de/data/flickr-growth.txt.gz ../data/
python3 flickr_process.py # process flickr-growth.txt.gz, which is downloaded from http://socialnetworks.mpi-sws.mpg.de/data/flickr-growth.txt.gz
python3 flickr_train.py

# download mag_cli.csv by google drive
python3 mag_process.py  # process mag_cli.csv, which is downloaded from https://drive.google.com/file/d/17bgLs1iR96JW3Rd0mex3IK8qyU-qRElB/view?usp=sharing
python3 mag_train.py

Cite

@article{ma2022fast,
  title={Fast Learning of MNL Model From General Partial Rankings with Application to Network Formation Modeling},
  author={Ma, Jiaqi and Zhang, Xingjian and Mei, Qiaozhu},
  journal={Proceedings of the 15th ACM International Conference on Web Search and Data Mining},
  year={2022}
}

Fast Learning of MNL Model From General Partial Rankings with Application to Network Formation Modeling

Related tags

Overview

Fast-Partial-Ranking-MNL

Requirements

Example Commands to Run the Experiments

Cite

Owner

Xingjian Zhang

Bootstrapped Unsupervised Sentence Representation Learning (ACL 2021)

[CVPR'21 Oral] Seeing Out of tHe bOx: End-to-End Pre-training for Vision-Language Representation Learning

JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library

Physics-informed Neural Operator for Learning Partial Differential Equation

SwinTrack: A Simple and Strong Baseline for Transformer Tracking

Cookiecutter PyTorch Lightning

This repository implements and evaluates convolutional networks on the Möbius strip as toy model instantiations of Coordinate Independent Convolutional Networks.

Deepparse is a state-of-the-art library for parsing multinational street addresses using deep learning

pybaum provides tools to work with pytrees which is a concept burrowed from JAX.

This is the implementation of "SELF SUPERVISED REPRESENTATION LEARNING WITH DEEP CLUSTERING FOR ACOUSTIC UNIT DISCOVERY FROM RAW SPEECH" submitted to ICASSP 2022

Source code of SIGIR2021 Paper 'One Chatbot Per Person: Creating Personalized Chatbots based on Implicit Profiles'

A multilingual version of MS MARCO passage ranking dataset

My personal Home Assistant configuration.

Dataset for the Research2Clinics @ NeurIPS 2021 Paper: What Do You See in this Patient? Behavioral Testing of Clinical NLP Models

U2-Net: Going Deeper with Nested U-Structure for Salient Object Detection

How the Deep Q-learning method works and discuss the new ideas that makes the algorithm work

Project Tugas Besar pertama Pengenalan Komputasi Institut Teknologi Bandung

StarGAN - Official PyTorch Implementation (CVPR 2018)

Contrastive learning of Class-agnostic Activation Map for Weakly Supervised Object Localization and Semantic Segmentation (CVPR 2022)

Rethinking Nearest Neighbors for Visual Classification