arXiv · 2305.19999
Beam Tree Recursive Cells
Abstract
We propose Beam Tree Recursive Cell (BT-Cell) - a backpropagation-friendly framework to extend Recursive Neural Networks (RvNNs) with beam search for latent structure induction. We further extend this framework by proposing a relaxation of the hard top-k operators in beam search for better propagation of gradient signals. We evaluate our proposed models in different out-of-distribution splits in both synthetic and realistic data. Our experiments show that BTCell achieves near-perfect performance on several challenging structure-sensitive synthetic tasks like ListOps and logical inference while maintaining comparable performance in realistic data against other RvNN-based models. Additionally, we identify a previously unknown failure case for neural models in generalization to unseen number of arguments in ListOps. The code is available at: https://github.com/JRC1995/BeamTreeRecursiveCells.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Jishnu Ray Chowdhury, Cornelia Caragea. 2023-05-31. Beam Tree Recursive Cells. https://arxiv.org/abs/2305.19999
Cite the original work for its findings. Save a collection to share your selection of sources.