Sascha Marton

Sascha Marton

Machine Learning Researcher and PhD Candidate

Publications

Listed below are my most relevant publications. You can find a complete list of my articles on my Google Scholar profile.

Conference Papers

Mitigating Information Loss in Tree-Based Reinforcement Learning via Direct Optimization

Published in International Conference on Learning Representations (Spotlight), 2025

We propose a novel method for symbolic RL that enables end-to-end gradient-based learning of interpretable, axis-aligned decision trees, combining policy gradient optimization with symbolic decision-making.

Recommended citation: Marton, Sascha, et al. (2025). "Mitigating Information Loss in Tree-Based Reinforcement Learning via Direct Optimization." The Thirteenth International Conference on Learning Representations. 1(1).
Download Paper | Download Slides

Decision Trees That Remember: Gradient-Based Learning of Recurrent Decision Trees with Memory

Published in New Frontiers in Associative Memory workshop at ICLR 2025, 2025

We propose ReMeDe trees, a recurrent decision tree architecture with internal memory, enabling efficient learning for sequential data through hard, axis-aligned decision rules trained via gradient descent.

Recommended citation: Marton, Sascha, et al. (2025). "Decision Trees That Remember: Gradient-Based Learning of Recurrent Decision Trees with Memory." New Frontiers in Associative Memory workshop at ICLR 2025. 1(1).
Download Paper

Journal Articles