Subgoal Search For Complex Reasoning Tasks. - Konrad Czechowski; Tomasz Odrzygózdz; Marek Zbysinski; Michal Zawalski; Krzysztof Olejnik; Yuhuai Wu; Lukasz Kucinski; Piotr Milos (NeurIPS, 2021)

Off-Policy Correction For Multi-Agent Reinforcement Learning. - Michal Zawalski; Blazej Osinski; Henryk Michalewski; Piotr Milos (AAMAS, 2022)

Fast and Precise: Adjusting Planning Horizon with Adaptive Subgoal Search. - Michal Zawalski; Michal Tyrolski; Konrad Czechowski; Tomasz Odrzygózdz; Damian Stachura; Piotr Piekos; Yuhuai Wu; Lukasz Kucinski; Piotr Milos (ICLR, 2023)

Robotic Control via Embodied Chain-of-Thought Reasoning. - Michal Zawalski; William Chen; Karl Pertsch; Oier Mees; Chelsea Finn; Sergey Levine (CoRL, 2024)

What Matters in Hierarchical Search for Combinatorial Reasoning Problems? - Michal Zawalski; Gracjan Góral; Michal Tyrolski; Emilia Wisnios; Franciszek Budrowski; Lukasz Kucinski; Piotr Milos (CoRR, 2024)

Contrastive Representations for Temporal Reasoning. - Alicja Ziarko; Michal Bortkiewicz; Michal Zawalski; Benjamin Eysenbach; Piotr Milos (CoRR, 2025)