3 papers
cs.CV2026
HeatKV: Head-tuned KV-cache Compression for Visual Autoregressive Modeling
Jonathan Cederlund, Axel Berg, William Isaksson +3
Visual Autoregressive (VAR) models have recently demonstrated impressive image generation quality while maintaining low latency. However, they suffer from severe KV-cache memory co…
cs.LG2025
Towards Federated Learning with On-device Training and Communication in 8-bit Floating Point
Bokun Wang, Axel Berg, Durmus Alp Emre Acar +1
Recent work has shown that 8-bit floating point (FP8) can be used for efficiently training neural networks with reduced computational cost compared to training in FP32/FP16. In thi…
eess.AS2024
wav2pos: Sound Source Localization using Masked Autoencoders
Axel Berg, Jens Gulin, Mark O'Connor +3
We present a novel approach to the 3D sound source localization task for distributed ad-hoc microphone arrays by formulating it as a set-to-set regression problem. By training a mu…