Publications

Object-centric causal representation learning
Posterior samples of source galaxies in strong gravitational lenses with score-based priors
Inferring accurate posteriors for high-dimensional representations of the brightness of gravitationally-lensed sources is a major challenge,… (voir plus) in part due to the difficulties of accurately quantifying the priors. Here, we report the use of a score-based model to encode the prior for the inference of undistorted images of background galaxies. This model is trained on a set of high-resolution images of undistorted galaxies. By adding the likelihood score to the prior score and using a reverse-time stochastic differential equation solver, we obtain samples from the posterior. Our method produces independent posterior samples and models the data almost down to the noise level. We show how the balance between the likelihood and the prior meet our expectations in an experiment with out-of-distribution data.
Bayesian learning of Causal Structure and Mechanisms with GFlowNets and Variational Bayes
Bayesian causal structure learning aims to learn a posterior distribution over directed acyclic graphs (DAGs), and the mechanisms that defin… (voir plus)e the relationship between parent and child variables. By taking a Bayesian approach, it is possible to reason about the uncertainty of the causal model. The notion of modelling the uncertainty over models is particularly crucial for causal structure learning since the model could be unidentifiable when given only a finite amount of observational data. In this paper, we introduce a novel method to jointly learn the structure and mechanisms of the causal model using Variational Bayes, which we call Variational Bayes-DAG-GFlowNet (VBG). We extend the method of Bayesian causal structure learning using GFlowNets to learn not only the posterior distribution over the structure, but also the parameters of a linear-Gaussian model. Our results on simulated data suggest that VBG is competitive against several baselines in modelling the posterior over DAGs and mechanisms, while offering several advantages over existing methods, including the guarantee to sample acyclic graphs, and the flexibility to generalize to non-linear causal mechanisms.
A circulating proteome-informed prognostic model of COVID-19 disease activity that relies on routinely available clinical laboratories
Karine Tremblay
Simon Rousseau
A minority of people infected with SARS-CoV-2 will develop severe COVID-19 disease. To help physicians predict who is more likely to require… (voir plus) admission to ICU, we conducted an unsupervised stratification of the circulating proteome that identified six endophenotypes (EPs) among 731 SARS-CoV-2 PCR-positive hospitalized participants in the Biobanque Québécoise de la COVID-19, with varying degrees of disease severity and times to intensive care unit (ICU) admission. One endophenotype, EP6, was associated with a greater proportion of ICU admission, ventilation support, acute respiratory distress syndrome (ARDS) and death. Clinical features of EP6 included increased levels of C-reactive protein, D-dimers, interleukin-6, ferritin, soluble fms-like tyrosine kinase-1, elevated neutrophils, and depleted lymphocytes, whereas another endophenotype (EP5) was associated with cardiovascular complications, congruent with elevated blood biomarkers of cardiovascular disease like N-terminal pro B-type natriuretic peptide (NT-proBNP), Growth Differentiation Factor-15 (GDF-15), and Troponin T. Importantly, a prognostic model solely based on clinical laboratory measurements was developed and validated on 903 patients that generalizes the EPs to new patients recruited across all pandemic waves (2020-2022) and create new opportunities for automated identification of high-risk groups in the clinic. Thus, this novel way to address pathogenesis that leverages detailed phenotypic information but relies on routinely available information in the clinic to favor translation may find applications in other diseases beyond COVID-19.
Representational ethical model calibration
Robert Carruthers
Isabel Straw
James K. Ruffle
Daniel Herron
Amy Nelson
Delmiro Fernandez-Reyes
Geraint Rees
Parashkev Nachev
Spectral Regularization: an Inductive Bias for Sequence Modeling
Adult neurogenesis acts as a neural regularizer
Lina M. Tran
Adam Santoro
Lulu Liu
Sheena A. Josselyn
Blake A. Richards
Paul W. Frankland
New neurons are continuously generated in the subgranular zone of the dentate gyrus throughout adulthood. These new neurons gradually integr… (voir plus)ate into hippocampal circuits, forming new naive synapses. Viewed from this perspective, these new neurons may represent a significant source of “wiring” noise in hippocampal networks. In machine learning, such noise injection is commonly used as a regularization technique. Regularization techniques help prevent overfitting training data and allow models to generalize learning to new, unseen data. Using a computational modeling approach, here we ask whether a neurogenesis-like process similarly acts as a regularizer, facilitating generalization in a category learning task. In a convolutional neural network (CNN) trained on the CIFAR-10 object recognition dataset, we modeled neurogenesis as a replacement/turnover mechanism, where weights for a randomly chosen small subset of hidden layer neurons were reinitialized to new values as the model learned to categorize 10 different classes of objects. We found that neurogenesis enhanced generalization on unseen test data compared to networks with no neurogenesis. Moreover, neurogenic networks either outperformed or performed similarly to networks with conventional noise injection (i.e., dropout, weight decay, and neural noise). These results suggest that neurogenesis can enhance generalization in hippocampal learning through noise injection, expanding on the roles that neurogenesis may have in cognition.
Automatic measure and normalization of spinal cord cross-sectional area using the pontomedullary junction
Julien Cohen‐Adad
Spinal cord cross-sectional area (CSA) is a relevant biomarker to assess spinal cord atrophy in neurodegenerative diseases. However, the con… (voir plus)siderable inter-subject variability among healthy participants currently limits its usage. Previous studies explored factors contributing to the variability, yet the normalization models required manual intervention and used vertebral levels as a reference, which is an imprecise prediction of the spinal levels. In this study we implemented a method to measure CSA automatically from a spatial reference based on the central nervous system (the pontomedullary junction, PMJ), we investigated factors to explain variability, and developed normalization strategies on a large cohort (N = 804). Following automatic spinal cord segmentation, vertebral labeling and PMJ labeling, the spinal cord CSA was computed on T1w MRI scans from the UK Biobank database. The CSA was computed using two methods. For the first method, the CSA was computed at the level of the C2–C3 intervertebral disc. For the second method, the CSA was computed at 64 mm caudally from the PMJ, this distance corresponding to the average distance between the PMJ and the C2–C3 disc across all participants. The effect of various demographic and anatomical factors was explored, and a stepwise regression found significant predictors; the coefficients of the best fit model were used to normalize CSA. CSA measured at C2–C3 disc and using the PMJ differed significantly (paired t-test, p-value = 0.0002). The best normalization model included thalamus, brain volume, sex and the interaction between brain volume and sex. The coefficient of variation went down for PMJ CSA from 10.09 (without normalization) to 8.59%, a reduction of 14.85%. For CSA at C2–C3, it went down from 9.96 to 8.42%, a reduction of 15.13 %. This study introduces an end-to-end automatic pipeline to measure and normalize cord CSA from a neurological reference. This approach requires further validation to assess atrophy in longitudinal studies. The inter-subject variability of CSA can be partly accounted for by demographics and anatomical factors.
BERTPerf: Inference Latency Predictor for BERT on ARM big.LITTLE Multi-Core Processors
M. Abdelgawad
S. H. Mozafari
J. J. Clark
B. H. Meyer
W. J. Gross
Hardware-aware Neural Architecture Search (NAS) and mapping & scheduling optimization methods are being used to find efficient implementatio… (voir plus)ns of computationally-intense language models such as BERT. This requires measuring real hardware inference latency: good design decisions simply cannot be made with proxy metrics such as FLOPs or the number of parameters. However, the time required to perform on-device latency measurements is prohibitive (e.g., a few days to a few weeks over the course of an optimization run). To address this, we present BERTPerf, a low-cost, highly-accurate method to predict the inference time of BERT on ARM big.LITTLE multi-core processors. BERTPerf exploits latency patterns at the layer-level to reduce on-device latency measurements, and captures the effect of caching and intermediate tensor allocations to reduce latency prediction error. BERTPerf reduces the maximum prediction error by 7–11% compared to the state-of-the-art, and requires 75% less on-device measurements compared to existing work at the same prediction error.
A General-Purpose Neural Architecture for Geospatial Systems
Nasim Rahaman
Francesco Locatello
Alexandre Lacoste
Li Erran Li
Bernhard Schölkopf
Active Keyword Selection to Track Evolving Topics on Twitter
How can we study social interactions on evolving topics at a mass scale? Over the past decade, researchers from diverse fields such as econo… (voir plus)mics, political science, and public health have often done this by querying Twitter's public API endpoints with hand-picked topical keywords to search or stream discussions. However, despite the API's accessibility, it remains difficult to select and update keywords to collect high-quality data relevant to topics of interest. In this paper, we propose an active learning method for rapidly refining query keywords to increase both the yielded topic relevance and dataset size. We leverage a large open-source COVID-19 Twitter dataset to illustrate the applicability of our method in tracking Tweets around the key sub-topics of Vaccine, Mask, and Lockdown. Our experiments show that our method achieves an average topic-related keyword recall 2x higher than baselines. We open-source our code along with a web interface for keyword selection to make data collection from Twitter more systematic for researchers.
Consistent Training via Energy-Based GFlowNets for Modeling Discrete Joint Distributions
Moksh J. Jain
Payel Das
Generative Flow Networks (GFlowNets) have demonstrated significant performance improvements for generating diverse discrete objects …