This is the code base for Identifying Selections for Unsupervised Subtask Discovery (Qiu, Yiwen & Zheng, Yujia & Zhang, Kun. NeurIPS, 2024).
We refer to separate subfolders for instructions:
phase_1_CItests -> see run_test_ci.sh
phase_2_seqNMF -> see README.md
phase_3_hierIL -> see MetaIL_Kitchen/script.sh
This repository contains codebases from:
- https://github.com/FeeLab/seqNMF [1]
- https://github.com/LucasCJYSDL/Multi-task-Hierarchical-AIRL [2]
[1] Mackevicius, Emily L., Andrew H. Bahle, Alex H. Williams, Shijie Gu, Natalia I. Denisenko, Mark S. Goldman, and Michale S. Fee. "Unsupervised discovery of temporal sequences in high-dimensional datasets, with applications to neuroscience." Elife 8 (2019): e38471.
[2] Chen, Jiayu, Dipesh Tamboli, Tian Lan, and Vaneet Aggarwal. "Multi-task hierarchical adversarial inverse reinforcement learning." In International Conference on Machine Learning, pp. 4895-4920. PMLR, 2023.