← Search

Torsten Scholak

5 accepted papers

2026

DiffuMamba: High-Throughput Diffusion LMs with Mamba Backbone

ICML 2026poster

Diffusion language models (DLMs) have emerged as a promising alternative to autoregressive (AR) generation, yet their reliance on Transformer backbones limits inference efficiency due to quadratic attention or KV-cache overhead. We introduce DiffuMamba, a masked diffusion language model built on a b…

Cited by 0SourceScholar
2025

BigDocs: An Open Dataset for Training Multimodal Models on Document and Code Tasks

ICLR 2025poster

Multimodal AI has the potential to significantly enhance document-understanding tasks, such as processing receipts, understanding workflows, extracting data from documents, and summarizing reports. Code generation tasks that require long-structured outputs can also be enhanced by multimodality. Desp…

Cited by 0SourcePDFScholar
2022

UnifiedSKG: Unifying and Multi-Tasking Structured Knowledge Grounding with Text-to-Text Language Models

EMNLP 2022main

Structured knowledge grounding (SKG) leverages structured knowledge to complete user requests, such as semantic parsing over databases and question answering over knowledge bases. Since the inputs and outputs of SKG tasks are heterogeneous, they have been studied separately by different communities,…

2021

DuoRAT: Towards Simpler Text-to-SQL Models

NAACL 2021long

Recent neural text-to-SQL models can effectively translate natural language questions to corresponding SQL queries on unseen databases. Working mostly on the Spider dataset, researchers have proposed increasingly sophisticated solutions to the problem. Contrary to this trend, in this paper we focus…

2021

PICARD: Parsing Incrementally for Constrained Auto-Regressive Decoding from Language Models

EMNLP 2021main

Large pre-trained language models for textual data have an unconstrained output space; at each decoding step, they can produce any of 10,000s of sub-word tokens. When fine-tuned to target constrained formal languages like SQL, these models often generate invalid code, rendering it unusable. We propo…