Transformer-based Hierarchical Clustering

This repository contains the code for the paper "Transformer-Based Hierarchical Clustering for Brain Network Analysis," to be published at IEEE-ISBI 2023.

Dataset

ABIDE can be accessed from here. The preprocessing scripts can be found at util/abide, with instructions in the same folder.

Usage

ABIDE

python main.py --config_filename setting/abide_dec_ortho_learnable_newpool_sftmx_topk_90_4_hierarchical_sum_5-4lr.yaml

ABCD

python main.py --config_filename setting/abcd_dec_ortho_learnable_newpool_sftmx_topk_90_4_hierarchical_sum.yaml