paper-with-me

Papers

Trellis Networks for Sequence Modeling

2018-10-15 · ICLR 2019 5 · Shaojie Bai, J. Zico Kolter, Vladlen Koltun

We present trellis networks, a new architecture for sequence modeling. On the one hand, a trellis network is a temporal convolutional network with special structure, characterized by weight tying across depth and direct injection of the input into deep layers. On the other hand, we show that truncated recurrent networks are equivalent to trellis networks with special sparsity structure in their weight matrices. Thus trellis networks with general weight matrices generalize truncated recurrent networks. We leverage these connections to design high-performing trellis networks that absorb structural and algorithmic elements from both recurrent and convolutional models. Experiments demonstrate that trellis networks outperform the current state of the art methods on a variety of challenging benchmarks, including word-level language modeling and character-level language modeling tasks, and stress tests designed to evaluate long-term memory retention. The code is available at https://github.com/locuslab/trellisnet .

📄 PDF Abstract BibTeX arXiv:1810.06682

Code (1)

locuslab/trellisnet 공식 구현 pytorch

Tasks

Language ModelingLanguage ModellingSequential Image Classification

Methods 이 논문이 사용한 방법론

Weight Tying Weight Tying improves the performance of language models by tying (sharing) the weights of the embedding and softmax layers. This…

Similar Papers 제목 키워드 기반

Trellis: Learning to Compress Key-Value Memory in Attention Models

2025-12-29 · Mahdi Karami, Ali Behrouz, Praneeth Kacham, Vahab Mirrokni arxiv

Transformers, while powerful, suffer from quadratic computational complexity and the ever-growing Key-Value (KV) cache of the attention mechanism. This paper introduces Trellis, a novel Transformer architecture with boun…

Online Learning of Trellis Diagram Using Neural Network for Robust Detection and Decoding

2022-02-22 · Jie Yang, Qinghe Du, Yi Jiang

This paper studies machine learning-assisted maximum likelihood (ML) and maximum a posteriori (MAP) receivers for a communication system with memory, which can be modelled by a trellis diagram. The prerequisite of the ML…

Data-Driven Symbol Detection for Intersymbol Interference Channels with Bursty Impulsive Noise

2024-05-17 · Boris Karanov, Chin-Hung Chen, Yan Wu, Alex Young 외

We developed machine learning approaches for data-driven trellis-based soft symbol detection in coded transmission over intersymbol interference (ISI) channels in presence of bursty impulsive noise (IN), for example enco…

Mitigating Nonlinear Interference by Limiting Energy Variations in Sphere Shaping

2022-01-03 · Yunus Can Gültekin, Alex Alvarado, Olga Vassilieva, Inwoong Kim 외

Band-trellis enumerative sphere shaping is proposed to decrease the energy variations in channel input sequences. Against sphere shaping, 0.74 dB SNR gain and up to 9% increase in data rates are demonstrated for single-s…

Deep Equilibrium Models

2019-09-03 · NeurIPS 2019 12 · Shaojie Bai, J. Zico Kolter, Vladlen Koltun

We present a new approach to modeling sequential data: the deep equilibrium model (DEQ). Motivated by an observation that the hidden layers of many existing deep sequence models converge towards some fixed point, we prop…

Language ModelingLanguage Modelling