Koji Hukushima

dblp:16/380 · DBLP profile ↗
← Back
2ranked-venue papers
0as first author
2since 2021 · last 2024
0000-0003-1153-1758ORCID · reported

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 1 · 1 since 2021Theory of computation · 1 · 1 since 2021
YearPublicationVenuePosition
2024 Learning Dynamics in Linear VAE: Posterior Collapse Threshold, Superfluous Latent Space Pitfalls, and Speedup with KL Annealing
abstract
Variational autoencoders (VAEs) face a notorious problem wherein the variational posterior often aligns closely with the prior, a phenomenon known as posterior collapse, which hinders the quality of representation learning. To mitigate this problem, an adjustable hyperparameter $\beta$ and a strategy for annealing this parameter, called KL annealing, are proposed. This study presents a theoretical analysis of the learning dynamics in a minimal VAE. It is rigorously proved that the dynamics converge to a deterministic process within the limit of large input dimensions, thereby enabling a detailed dynamical analysis of the generalization error. Furthermore, the analysis shows that the VAE initially learns entangled representations and gradually acquires disentangled representations. A fixed-point analysis of the deterministic process reveals that when $\beta$ exceeds a certain threshold, posterior collapse becomes inevitable regardless of the learning period. Additionally, the superfluous latent variables for the data-generative factors lead to overfitting of the background noise; this adversely affects both generalization and learning convergence. The analysis further unveiled that appropriately tuned KL annealing can accelerate convergence.
Yuma Ichikawa, Koji Hukushima
AISTATS2
2024 Adaptive Flip Graph Algorithm for Matrix Multiplication
abstract
This study proposes the “adaptive flip graph algorithm”, which combines adaptive searches with the flip graph algorithm for finding fast and efficient methods for matrix multiplication. The adaptive flip graph algorithm addresses the inherent limitations of exploration and inefficient search encountered in the original flip graph algorithm, particularly when dealing with large matrix multiplication. For the limitation of exploration, the proposed algorithm adaptively transitions over the flip graph, introducing a flexibility that does not strictly reduce the number of multiplications. Concerning the issue of inefficient search in large instances, the proposed algorithm adaptively constraints the search range instead of relying on a completely random search, facilitating more effective exploration. In particular, a formal proof is provided that the introduction of plus transitions in the proposed algorithm ensures the connectivity of any node in the flip graph, which represents a method of matrix multiplication. Numerical experimental results demonstrate the effectiveness of the adaptive flip graph algorithm, which involves applying matrices calculated in characteristic 2. This algorithm reduces the number of multiplications for a 4 × 5 matrix multiplied by a 5 × 5 matrix from 76 to 73 and that for a 5 × 5 matrix multiplied by another 5 × 5 matrix from 95 to 94.
Yamato Arai, Yuma Ichikawa, Koji Hukushima
ISSAC3