new

Get trending papers in your email inbox!

Subscribe

Daily Papers

byAK and the research community

Oct 2

SPT: Fine-Tuning Transformer-based Language Models Efficiently with Sparsification

Transformer-based large language models (e.g., BERT and GPT) achieve great success, and fine-tuning, which tunes a pre-trained model on a task-specific dataset, is the standard practice to utilize these models for downstream tasks. However, Transformer fine-tuning has long running time and high memory consumption due to the large size of the models. We propose the SPT system to fine-tune Transformer-based models efficiently by introducing sparsity. We observe that the memory consumption of Transformer mainly comes from storing attention weights for multi-head attention (MHA), and the majority of running time is spent on feed-forward network (FFN). Thus, we design the sparse MHA module, which computes and stores only large attention weights to reduce memory consumption, and the routed FFN module, which dynamically activates a subset of model parameters for each token to reduce computation cost. We implement SPT on PyTorch and customize CUDA kernels to run sparse MHA and routed FFN efficiently. Specifically, we use product quantization to identify the large attention weights and compute attention via sparse matrix multiplication for sparse MHA. For routed FFN, we batch the tokens according to their activated model parameters for efficient computation. We conduct extensive experiments to evaluate SPT on various model configurations. The results show that SPT consistently outperforms well-optimized baselines, reducing the peak memory consumption by up to 50% and accelerating fine-tuning by up to 2.2x.

  • 5 authors
·
Dec 16, 2023 2

LM-SPT: LM-Aligned Semantic Distillation for Speech Tokenization

With the rapid progress of speech language models (SLMs), discrete speech tokens have emerged as a core interface between speech and text, enabling unified modeling across modalities. Recent speech tokenization approaches aim to isolate semantic information from low-level acoustics to better align with language models. In particular, previous methods use SSL teachers such as HuBERT to extract semantic representations, which are then distilled into a semantic quantizer to suppress acoustic redundancy as well as capture content-related latent structures. However, they still produce speech token sequences significantly longer than their textual counterparts, creating challenges for efficient speech-language modeling. Reducing the frame rate is a natural solution, but standard techniques, such as rigid average pooling across frames, can distort or dilute the semantic structure required for effective LM alignment. To address this, we propose LM-SPT, a speech tokenization method that introduces a novel semantic distillation. Instead of directly matching teacher and student features via pooling, we reconstruct speech solely from semantic tokens and minimize the discrepancy between the encoded representations of the original and reconstructed waveforms, obtained from a frozen automatic speech recognition (ASR) encoder. This indirect yet data-driven supervision enables the tokenizer to learn discrete units that are more semantically aligned with language models. LM-SPT further incorporates architectural improvements to the encoder and decoder for speech tokenization, and supports multiple frame rates, including 25Hz, 12.5Hz, and 6.25Hz. Experimental results show that LM-SPT achieves superior reconstruction fidelity compared to baselines, and that SLMs trained with LM-SPT tokens achieve competitive performances on speech-to-text and consistently outperform baselines on text-to-speech tasks.

  • 4 authors
·
Jun 20, 2025

The SPTpol Extended Cluster Survey

We describe the observations and resultant galaxy cluster catalog from the 2770 deg^2 SPTpol Extended Cluster Survey (SPT-ECS). Clusters are identified via the Sunyaev-Zel'dovich (SZ) effect, and confirmed with a combination of archival and targeted follow-up data, making particular use of data from the Dark Energy Survey (DES). With incomplete followup we have confirmed as clusters 244 of 266 candidates at a detection significance ξge 5 and an additional 204 systems at 4<ξ<5. The confirmed sample has a median mass of M_{500c} 4.4 times 10^{14 M_odot h_{70}^{-1}}, a median redshift of z=0.49, and we have identified 44 strong gravitational lenses in the sample thus far. Radio data are used to characterize contamination to the SZ signal; the median contamination for confirmed clusters is predicted to be sim1% of the SZ signal at the ξ>4 threshold, and <4% of clusters have a predicted contamination >10% of their measured SZ flux. We associate SZ-selected clusters, from both SPT-ECS and the SPT-SZ survey, with clusters from the DES redMaPPer sample, and find an offset distribution between the SZ center and central galaxy in general agreement with previous work, though with a larger fraction of clusters with significant offsets. Adopting a fixed Planck-like cosmology, we measure the optical richness-to-SZ-mass (λ-M) relation and find it to be 28% shallower than that from a weak-lensing analysis of the DES data---a difference significant at the 4 σ level---with the relations intersecting at λ=60 . The SPT-ECS cluster sample will be particularly useful for studying the evolution of massive clusters and, in combination with DES lensing observations and the SPT-SZ cluster sample, will be an important component of future cosmological analyses.

  • 138 authors
·
Dec 12, 2019

Galaxy Clusters Discovered via the Thermal Sunyaev-Zel&#39;dovich Effect in the 500-square-degree SPTpol Survey

We present a catalog of 689 galaxy cluster candidates detected at significance ξ>4 via their thermal Sunyaev-Zel'dovich (SZ) effect signature in 95 and 150 GHz data from the 500-square-degree SPTpol survey. We use optical and infrared data from the Dark Energy Camera and the Wide-field Infrared Survey Explorer (WISE) and \spitzer \ satellites, to confirm 544 of these candidates as clusters with sim94% purity. The sample has an approximately redshift-independent mass threshold at redshift z>0.25 and spans 1.5 times 10^{14} < M_{500c} < 9.1 times 10^{14} M_odot/h_{70} \ and 0.03<zlesssim1.6 in mass and redshift, respectively; 21\% of the confirmed clusters are at z>1. We use external radio data from the Sydney University Molonglo Sky Survey (SUMSS) to estimate contamination to the SZ signal from synchrotron sources. The contamination reduces the recovered ξ by a median value of 0.032, or sim0.8% of the ξ=4 threshold value, and sim7% of candidates have a predicted contamination greater than Δξ= 1. With the exception of a small number of systems (<1%), an analysis of clusters detected in single-frequency 95 and 150 GHz data shows no significant contamination of the SZ signal by emission from dusty or synchrotron sources. This cluster sample will be a key component in upcoming astrophysical and cosmological analyses of clusters. The SPTpol millimeter-wave maps and associated data products used to produce this sample are available at https://pole.uchicago.edu/public/data/sptpol_500d_clusters/index.html, and the NASA LAMBDA website. An interactive sky server with the SPTpol maps and Dark Energy Survey data release 2 images is also available at NCSA https://skyviewer.ncsa.illinois.edu.

  • 128 authors
·
Feb 7, 2024

Galaxy Clusters Selected via the Sunyaev-Zel&#39;dovich Effect in the SPTpol 100-Square-Degree Survey

We present a catalog of galaxy cluster candidates detected in 100 square degrees surveyed with the SPTpol receiver on the South Pole Telescope. The catalog contains 89 candidates detected with a signal-to-noise ratio greater than 4.6. The candidates are selected using the Sunyaev-Zel'dovich effect at 95 and 150 GHz. Using both space- and ground-based optical and infrared telescopes, we have confirmed 81 candidates as galaxy clusters. We use these follow-up images and archival images to estimate photometric redshifts for 66 galaxy clusters and spectroscopic observations to obtain redshifts for 13 systems. An additional 2 galaxy clusters are confirmed using the overdensity of near-infrared galaxies only, and are presented without redshifts. We find that 15 candidates (18% of the total sample) are at redshift of z geq 1.0, with a maximum confirmed redshift of z_{max} = 1.38 pm 0.10. We expect this catalog to contain every galaxy cluster with M_{500c} > 2.6 times 10^{14} M_odot h^{-1}_{70} and z > 0.25 in the survey area. The mass threshold is approximately constant above z = 0.25, and the complete catalog has a median mass of approximately M_{500c} = 2.7 times 10^{14} M_odot h^{-1}_{70}. Compared to previous SPT works, the increased depth of the millimeter-wave data (11.2 and 6.5 μK-arcmin at 95 and 150 GHz, respectively) makes it possible to find more galaxy clusters at high redshift and lower mass.

  • 77 authors
·
Jan 13, 2020

Measurements of the Temperature and E-mode Polarization of the Cosmic Microwave Background from the Full 500-square-degree SPTpol Dataset

Using the full four-year SPTpol 500 deg^2 dataset in both the 95 GHz and 150 GHz frequency bands, we present measurements of the temperature and E-mode polarization of the cosmic microwave background (CMB), as well as the E-mode polarization auto-power spectrum (EE) and temperature-E-mode cross-power spectrum (TE) in the angular multipole range 50<ell<8000. We find the SPTpol dataset to be self-consistent, passing several internal consistency tests based on maps, frequency bands, bandpowers, and cosmological parameters. The full SPTpol dataset is well-fit by the ΛCDM model, for which we find H_0=70.48pm2.16 km s^{-1} Mpc^{-1} and Ω_m=0.271pm0.026, when using only the SPTpol data and a Planck-based prior on the optical depth to reionization. The ΛCDM parameter constraints are consistent across the 95 GHz-only, 150 GHz-only, TE-only, and EE-only data splits. Between the ell<1000 and ell>1000 data splits, the ΛCDM parameter constraints are borderline consistent at the sim2σ level. This consistency improves when including a parameter A_L, the degree of lensing of the CMB inferred from the smearing of acoustic peaks. When marginalized over A_L, the ΛCDM parameter constraints from SPTpol are consistent with those from Planck. The power spectra presented here are the most sensitive measurements of the lensed CMB damping tail to date for roughly ell > 1700 in TE and ell > 2000 in EE.

  • 62 authors
·
Aug 1, 2025

Cluster Cosmology Constraints from the 2500 deg$^2$ SPT-SZ Survey: Inclusion of Weak Gravitational Lensing Data from Magellan and the Hubble Space Telescope

We derive cosmological constraints using a galaxy cluster sample selected from the 2500~deg^2 SPT-SZ survey. The sample spans the redshift range 0.25< z<1.75 and contains 343 clusters with SZ detection significance ξ>5. The sample is supplemented with optical weak gravitational lensing measurements of 32 clusters with 0.29<z<1.13 (from Magellan and HST) and X-ray measurements of 89 clusters with 0.25<z<1.75 (from Chandra). We rely on minimal modeling assumptions: i) weak lensing provides an accurate means of measuring halo masses, ii) the mean SZ and X-ray observables are related to the true halo mass through power-law relations in mass and dimensionless Hubble parameter E(z) with a-priori unknown parameters, iii) there is (correlated, lognormal) intrinsic scatter and measurement noise relating these observables to their mean relations. We simultaneously fit for these astrophysical modeling parameters and for cosmology. Assuming a flat νΛCDM model, in which the sum of neutrino masses is a free parameter, we measure Ω_m=0.276pm0.047, σ_8=0.781pm0.037, and σ_8(Ω_m/0.3)^{0.2}=0.766pm0.025. The redshift evolution of the X-ray Y_X-mass and M_gas-mass relations are both consistent with self-similar evolution to within 1σ. The mass-slope of the Y_X-mass relation shows a 2.3σ deviation from self-similarity. Similarly, the mass-slope of the M_gas-mass relation is steeper than self-similarity at the 2.5σ level. In a νwCDM cosmology, we measure the dark energy equation of state parameter w=-1.55pm0.41 from the cluster data. We perform a measurement of the growth of structure since redshift zsim1.7 and find no evidence for tension with the prediction from General Relativity. We provide updated redshift and mass estimates for the SPT sample. (abridged)

  • 80 authors
·
May 19, 2019