kth.sePublikationer KTH
Ändra sökning
RefereraExporteraLänk till posten
Permanent länk

Direktlänk
Referera
Referensformat
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Annat format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annat 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 (Engelska)Ingår i: Transactions on Machine Learning Research, E-ISSN 2835-8856, Vol. 2024, nr 09Artikel i tidskrift (Refereegranskat) 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.

Ort, förlag, år, upplaga, sidor
Transactions on Machine Learning Research , 2024. Vol. 2024, nr 09
Nationell ämneskategori
Beräkningsmatematik Annan data- och informationsvetenskap Datavetenskap (datalogi)
Identifikatorer
URN: urn:nbn:se:kth:diva-367210Scopus ID: 2-s2.0-85214744354OAI: oai:DiVA.org:kth-367210DiVA, id: diva2:1984369
Anmärkning

QC 20250715

Tillgänglig från: 2025-07-15 Skapad: 2025-07-15 Senast uppdaterad: 2025-07-15Bibliografiskt granskad

Open Access i DiVA

Fulltext saknas i DiVA

Övriga länkar

Scopusfulltext

Person

Öktem, Ozan

Sök vidare i DiVA

Av författaren/redaktören
Öktem, Ozan
Av organisationen
Numerisk analys, optimeringslära och systemteori
I samma tidskrift
Transactions on Machine Learning Research
BeräkningsmatematikAnnan data- och informationsvetenskapDatavetenskap (datalogi)

Sök vidare utanför DiVA

GoogleGoogle Scholar

urn-nbn

Altmetricpoäng

urn-nbn
Totalt: 149 träffar
RefereraExporteraLänk till posten
Permanent länk

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