Endre søk
RefereraExporteraLink to record
Permanent link

Direct link
Referera
Referensformat
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Annet format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annet språk
Fler språk
Utmatningsformat
  • html
  • text
  • asciidoc
  • rtf
Neural incomplete factorization: learning preconditioners for the conjugate gradient method
Department of Information Technology, Uppsala University, Sweden.
KTH, Skolan för teknikvetenskap (SCI), Matematik (Inst.), Numerisk analys, optimeringslära och systemteori.ORCID-id: 0000-0002-1118-6483
Department of Information Technology, Uppsala University, Sweden.
2024 (engelsk)Inngår i: Transactions on Machine Learning Research, E-ISSN 2835-8856, Vol. 2024, nr 09Artikkel i tidsskrift (Fagfellevurdert) Published
Abstract [en]

The convergence of the conjugate gradient method for solving large-scale and sparse linear equation systems depends on the spectral properties of the system matrix, which can be improved by preconditioning. In this paper, we develop a computationally efficient data-driven approach to accelerate the generation of effective preconditioners. We, therefore, replace the typically hand-engineered preconditioners by the output of graph neural networks. Our method generates an incomplete factorization of the matrix and is, therefore, referred to as neural incomplete factorization (NeuralIF). Optimizing the condition number of the linear system directly is computationally infeasible. Instead, we utilize a stochastic approximation of the Frobenius loss which only requires matrix-vector multiplications for efficient training. At the core of our method is a novel message-passing block, inspired by sparse matrix theory, that aligns with the objective of finding a sparse factorization of the matrix. We evaluate our proposed method on both synthetic problem instances and on problems arising from the discretization of the Poisson equation on varying domains. Our experiments show that by using data-driven preconditioners within the conjugate gradient method we are able to speed up the convergence of the iterative procedure. The code is available at https://github.com/paulhausner/neural-incomplete-factorization.

sted, utgiver, år, opplag, sider
Transactions on Machine Learning Research , 2024. Vol. 2024, nr 09
HSV kategori
Identifikatorer
URN: urn:nbn:se:kth:diva-367210Scopus ID: 2-s2.0-85214744354OAI: oai:DiVA.org:kth-367210DiVA, id: diva2:1984369
Merknad

QC 20250715

Tilgjengelig fra: 2025-07-15 Laget: 2025-07-15 Sist oppdatert: 2025-07-15bibliografisk kontrollert

Open Access i DiVA

Fulltekst mangler i DiVA

Andre lenker

Scopusfulltext

Person

Öktem, Ozan

Søk i DiVA

Av forfatter/redaktør
Öktem, Ozan
Av organisasjonen
I samme tidsskrift
Transactions on Machine Learning Research

Søk utenfor DiVA

GoogleGoogle Scholar

urn-nbn

Altmetric

urn-nbn
Totalt: 141 treff
RefereraExporteraLink to record
Permanent link

Direct link
Referera
Referensformat
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Annet format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annet språk
Fler språk
Utmatningsformat
  • html
  • text
  • asciidoc
  • rtf