Search Results for author: Yiheng Jiang

Found 2 papers, 1 papers with code

Algorithms for mean-field variational inference via polyhedral optimization in the Wasserstein space

no code implementations5 Dec 2023 Yiheng Jiang, Sinho Chewi, Aram-Alexandre Pooladian

We develop a theory of finite-dimensional polyhedral subsets over the Wasserstein space and optimization of functionals over them via first-order methods.

Variational Inference

TriMLP: Revenge of a MLP-like Architecture in Sequential Recommendation

1 code implementation24 May 2023 Yiheng Jiang, Yuanbo Xu, Yongjian Yang, Funing Yang, Pengyang Wang, Hui Xiong

In this paper, we present a MLP-like architecture for sequential recommendation, namely TriMLP, with a novel Triangular Mixer for cross-token communications.

Sequential Recommendation

Cannot find the paper you are looking for? You can Submit a new open access paper.