Cornell University
Library
Cornell UniversityLibrary

eCommons

Help
Log In(current)
  1. Home
  2. Cornell University Graduate School
  3. Cornell Theses and Dissertations
  4. STOCHASTIC OPTIMIZATION FOR DISTRIBUTED TRAINING AND MULTI-AGENT INTERACTION IN LARGE LANGUAGE MODELS

STOCHASTIC OPTIMIZATION FOR DISTRIBUTED TRAINING AND MULTI-AGENT INTERACTION IN LARGE LANGUAGE MODELS

File(s)
Jain_cornellgrad_0058F_15414.pdf (18.33 MB)
Permanent Link(s)
https://doi.org/10.7298/d3b9-2s58
https://hdl.handle.net/1813/126501
Collections
Cornell Theses and Dissertations
Author
Jain, Adit
Abstract

This thesis develops stochastic optimization frameworks for training and coordinating Large Language Models (LLMs) in distributed settings. We study two interconnected problems: (i) controlling the training dynamics of LLMs across distributed datasets, and (ii) modeling and controlling emergent behaviors when multiple LLM agents interact over networks. Part I formulates distributed LLM training as a controlled stochastic optimization problem. We model the training process as a Markov decision process (MDP) where a learner optimally schedules queries to a Markovian oracle providing noisy gradient evaluations. For covert optimization—where a learner must hide the optimal parameters from an eavesdropper—we prove the optimal policy exhibits a monotone threshold structure exploiting supermodularity of the dynamic programming operator. We extend this framework to multi-learner settings via constrained switching control games and develop policy gradient algorithms that provably converge to optimal threshold policies. Applications to privacy-preserving federated learning for hate speech classification demonstrate that our methods reduce eavesdropper accuracy by over 30% compared to greedy baselines. Part II develops analytical frameworks for networks of interacting LLM agents performing distributed inference. We model individual LLM agents as rationally inattentive Bayesian utility maximizers, providing interpretable microeconomic characterizations of their decision-making. For sequential Bayesian social learning, we characterize conditions under which LLM agents form information cascades (herds) and derive optimal stopping policies to delay herding and improve collective accuracy. For large-scale networks, we develop mean-field approximations that accurately predict population dynamics and enable principled network design via preferential attachment mechanisms. Empirical validation on networks of up to 100 LLMs confirms the predictive power of our theoretical framework. Part III presents additional independent contributions in online learning and game theory: (i) sublinear regret algorithms for blocked sparse linear bandits with applications to personalized recommendation and adaptive annotation; (ii) bandit algorithms for multimodal reward distributions with mixture-of-Gaussians structure; (iii) learning dynamics in asymmetric network games using $\alpha$-potential functions that guarantee convergence to approximate Nash equilibria; and (iv) mixture-of-tokens generation methods for reinforcement learning with verifiable feedback in LLMs.

Description
587 pages
Date Issued
2026-05
Keywords
covertness
•
large language models
•
network science
•
reinforcement learning
•
stochastic optimization
Committee Chair
Krishnamurthy, Vikram
Committee Member
Zhao, Qing
Tardos, Eva
Degree Discipline
Electrical and Computer Engineering
Degree Name
Ph. D., Electrical and Computer Engineering
Degree Level
Doctor of Philosophy
Rights
Attribution-NonCommercial-ShareAlike 4.0 International
Rights URI
https://creativecommons.org/licenses/by-nc-sa/4.0/
Type
dissertation or thesis

Site Statistics | Help

About eCommons | Policies | Terms of use | Contact Us

copyright © 2002-2026 Cornell University Library | Privacy | Web Accessibility Assistance